From a2db446cdb38eb0aefab814036e7f19866586e5e Mon Sep 17 00:00:00 2001 From: Macy Huston Date: Tue, 25 Nov 2025 11:52:35 -0800 Subject: [PATCH] Update model_fitter.py --- src/bagle/model_fitter.py | 12 +++++++++--- 1 file changed, 9 insertions(+), 3 deletions(-) diff --git a/src/bagle/model_fitter.py b/src/bagle/model_fitter.py index 70c957e..34c3e4f 100755 --- a/src/bagle/model_fitter.py +++ b/src/bagle/model_fitter.py @@ -944,7 +944,10 @@ def callback_plotter(self, nSamples, nlive, nPar, def get_modified_mag_err(self, cube, filt_index): mag_err = copy.deepcopy(self.data['mag_err' + str(filt_index + 1)]) - if self.add_error_on_photometry: + if (self.add_error_on_photometry and \ + not isinstance(self.add_error_on_photometry, (list, np.ndarray))) or \ + (isinstance(self.add_error_on_photometry, (list, np.ndarray)) \ + and self.add_error_on_photometry[filt_index]): add_err_name = 'add_err' + str(filt_index + 1) if isinstance(cube, dict) or isinstance(cube, Row): add_err = cube[add_err_name] @@ -953,7 +956,10 @@ def get_modified_mag_err(self, cube, filt_index): add_err = cube[add_err_idx] mag_err = np.hypot(mag_err, add_err) - if self.multiply_error_on_photometry: + if (self.multiply_error_on_photometry and \ + not isinstance(self.multiply_error_on_photometry, (list, np.ndarray))) or \ + (isinstance(self.multiply_error_on_photometry, (list, np.ndarray)) \ + and self.multiply_error_on_photometry[filt_index]): mult_err_name = 'mult_err' + str(filt_index + 1) if isinstance(cube, dict) or isinstance(cube, Row): mult_err = cube[mult_err_name] @@ -962,7 +968,7 @@ def get_modified_mag_err(self, cube, filt_index): mult_err = cube[mult_err_idx] mag_err *= mult_err - return mag_err + return mag_err def write_params_yaml(self):