more logging

cmu-delphi · Sep 9, 2024 · ed4ff19 · ed4ff19
1 parent 5481f0c
commit ed4ff19
Show file tree

Hide file tree

Showing 2 changed files with 4 additions and 0 deletions.
diff --git a/_delphi_utils_python/delphi_utils/export.py b/_delphi_utils_python/delphi_utils/export.py
@@ -129,4 +129,6 @@ def create_export_csv(
         if sort_geos:
             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)
     return dates
diff --git a/claims_hosp/delphi_claims_hosp/update_indicator.py b/claims_hosp/delphi_claims_hosp/update_indicator.py
@@ -216,6 +216,8 @@ def preprocess_output(self, df) -> pd.DataFrame:
         filtered_df.rename(columns={"rate": "val"}, inplace=True)
         filtered_df["timestamp"] = filtered_df["timestamp"].astype(str)
         df_list = []
+        if self.write_se:
+            logging.info("WARNING: WRITING SEs")
         for geo_id, group in filtered_df.groupby("geo_id"):
             assert not group.val.isnull().any()
             assert not group.se.isnull().any()