Edits to placement of logger argument

dtcenter · Sep 18, 2024 · 28640b3 · 28640b3
1 parent 5ef9795
commit 28640b3
Showing 1 changed file with 4 additions and 4 deletions.
diff --git a/metcalcpy/agg_stat_event_equalize.py b/metcalcpy/agg_stat_event_equalize.py
@@ -116,11 +116,11 @@ def calculate_values(self):
                     logger.debug(f"Series variable values for {key}: {all_series_vars[key]}")
                 logger.info("Running event equalization on all data.")
 
-                output_ee_data = event_equalize(all_ee_records, self.params['indy_var'],
+                output_ee_data = event_equalize(logger, all_ee_records, self.params['indy_var'],
                                                 all_series_vars,
                                                 list(self.params['fixed_vars_vals_input'].keys()),
                                                 fix_vals_permuted, True,
-                                                False, logger)
+                                                False)
                 logger.debug(f"Final event equalization completed. Number of records: {len(output_ee_data)}")
 
         else:
@@ -164,10 +164,10 @@ def run_ee_on_axis(self, fix_vals_permuted, axis='1'):
             # perform EE on filtered data
             logger.info(f"Performing event equalization on {series_var}.")
             series_data_after_ee = \
-                event_equalize(series_data_for_ee, self.params['indy_var'],
+                event_equalize(logger, series_data_for_ee, self.params['indy_var'],
                                self.params['series_val_' + axis],
                                list(self.params['fixed_vars_vals_input'].keys()),
-                               fix_vals_permuted, True, False, logger)
+                               fix_vals_permuted, True, False)
             logger.debug(f"Number of records after event equalization for {series_var}: {len(series_data_after_ee)}")
 
             # append EE data to result