Fix bug concerning uppercase arguments and click

Author: tobias-liaudat

long-runs/train_eval_plot_script_click_multi_cycle.py

@@ -369,13 +369,13 @@
     help="Whether extrapolation is performed or not on the borders of the SED."
 )
 @click.option(
-    "--SED_interp_kind",
+    "--sed_interp_kind",
     default="linear",
     type=str,
     help="Type of interpolation for the SED."
 )
 @click.option(
-    "--SED_sigma",
+    "--sed_sigma",
     default=0,
     type=float,
     help="Standard deviation of the multiplicative SED Gaussian noise."

wf_psf/script_utils.py

@@ -108,7 +108,7 @@ def train_model(**args):
     if 'interp_pts_per_bin' not in args:
         args['interp_pts_per_bin'] = 0
         args['extrapolate'] = True
-        args['SED_interp_kind'] = 'linear'
+        args['sed_interp_kind'] = 'linear'
 
     ## Generate initializations
     # Prepare np input
@@ -121,8 +121,8 @@ def train_model(**args):
         output_Q=args['output_q'],
         interp_pts_per_bin=args['interp_pts_per_bin'],
         extrapolate=args['extrapolate'],
-        SED_interp_kind=args['SED_interp_kind'],
-        SED_sigma=args['SED_sigma']
+        SED_interp_kind=args['sed_interp_kind'],
+        SED_sigma=args['sed_sigma']
     )
     simPSF_np.gen_random_Z_coeffs(max_order=args['n_zernikes'])
     z_coeffs = simPSF_np.normalize_zernikes(simPSF_np.get_z_coeffs(), simPSF_np.max_wfe_rms)
@@ -597,7 +597,7 @@ def evaluate_model(**args):
     if 'interp_pts_per_bin' not in args:
         args['interp_pts_per_bin'] = 0
         args['extrapolate'] = True
-        args['SED_interp_kind'] = 'linear'
+        args['sed_interp_kind'] = 'linear'
 
     # Prepare np input
     simPSF_np = SimPSFToolkit(
@@ -609,8 +609,8 @@ def evaluate_model(**args):
         output_Q=args['output_q'],
         interp_pts_per_bin=args['interp_pts_per_bin'],
         extrapolate=args['extrapolate'],
-        SED_interp_kind=args['SED_interp_kind'],
-        SED_sigma=args['SED_sigma']
+        SED_interp_kind=args['sed_interp_kind'],
+        SED_sigma=args['sed_sigma']
     )
     simPSF_np.gen_random_Z_coeffs(max_order=args['n_zernikes'])
     z_coeffs = simPSF_np.normalize_zernikes(simPSF_np.get_z_coeffs(), simPSF_np.max_wfe_rms)