diff --git a/_delphi_utils_python/delphi_utils/export.py b/_delphi_utils_python/delphi_utils/export.py index ef275c708..3157254cc 100644 --- a/_delphi_utils_python/delphi_utils/export.py +++ b/_delphi_utils_python/delphi_utils/export.py @@ -130,5 +130,5 @@ def create_export_csv( export_df = export_df.sort_values(by="geo_id") export_df.to_csv(export_file, index=False, na_rep="NA") - logger.debug("Wrote rows", num_rows=df.size, geo_type=geo_level, num_geo_ids=export_df["geo_id"].unique().size) + logger.debug("Wrote rows", num_rows=df.size, geo_type=geo_res, num_geo_ids=export_df["geo_id"].unique().size) return dates diff --git a/claims_hosp/delphi_claims_hosp/run.py b/claims_hosp/delphi_claims_hosp/run.py index 6128ee46a..25d74f901 100644 --- a/claims_hosp/delphi_claims_hosp/run.py +++ b/claims_hosp/delphi_claims_hosp/run.py @@ -146,6 +146,7 @@ def run_module(params): start_date=startdate, geo_res=geo, sensor=signal_name, + logger=logger ) max_dates.append(updater.output_dates[-1]) diff --git a/claims_hosp/tests/test_update_indicator.py b/claims_hosp/tests/test_update_indicator.py index 3a83ad1bb..49433a599 100644 --- a/claims_hosp/tests/test_update_indicator.py +++ b/claims_hosp/tests/test_update_indicator.py @@ -171,7 +171,8 @@ def test_write_to_csv_results(self): start_date=self.start_date, end_date=self.end_date, geo_res=res0["geo_level"], - sensor=Config.signal_name) + sensor=Config.signal_name, + logger=TEST_LOGGER) # check outputs expected_name = f"20200501_geography_{Config.signal_name}.csv" assert exists(join(td.name, expected_name)) @@ -255,7 +256,8 @@ def test_write_to_csv_with_se_results(self): start_date=self.start_date, end_date=self.end_date, geo_res=res0["geo_level"], - sensor=signal_name) + sensor=signal_name, + logger=TEST_LOGGER) # check outputs expected_name = f"20200501_geography_{signal_name}.csv"