protect against missing stuff in config

segasai · Nov 18, 2024 · d05af18 · d05af18
1 parent 3d64465
commit d05af18
Showing 1 changed file with 4 additions and 4 deletions.
diff --git a/py/rvspecfit/desi/desi_fit.py b/py/rvspecfit/desi/desi_fit.py
@@ -1103,11 +1103,11 @@ def proc_desi(fname,
         sig0s = {}
         for s in setups:
             if config is not None:
-                if 'lsf_sigma0_angstrom' not in config:
+                if ('lsf_sigma0_angstrom' not in config
+                        or s not in config['lsf_sigma0_angstrom']):
                     cur_val = 0.5  # default value
-                    logging.warning(
-                        'sigma0 of the templates is not specified, '
-                        f'using {cur_val}')
+                    logging.warning('sigma0 of the templates is not specified '
+                                    f'for setup {s} using {cur_val}')
                 else:
                     cur_val = config['lsf_sigma0_angstrom'][s]
                 sig0s[s] = cur_val