diff --git a/nssp/delphi_nssp/constants.py b/nssp/delphi_nssp/constants.py index ef7acb4f7..23c62c3f5 100644 --- a/nssp/delphi_nssp/constants.py +++ b/nssp/delphi_nssp/constants.py @@ -9,7 +9,7 @@ "state", "county", "hhs", - "hsa-nci", + "hsa_nci", ] SIGNALS_MAP = { diff --git a/nssp/delphi_nssp/run.py b/nssp/delphi_nssp/run.py index dc85055e5..418de0963 100644 --- a/nssp/delphi_nssp/run.py +++ b/nssp/delphi_nssp/run.py @@ -138,7 +138,7 @@ def run_module(params, logger=None): df = geo_mapper.add_geocode(df, "state_code", "hhs", from_col="state_code", new_col="geo_id") df = geo_mapper.aggregate_by_weighted_sum(df, "geo_id", "val", "timestamp", "population") df = df.rename(columns={"weighted_val": "val"}) - elif geo == "hsa-nci": + elif geo == "hsa_nci": df = df[["hsa_nci_id", "val", "timestamp"]] df = df[df["hsa_nci_id"] != "All"] # We use drop_duplicates below just to pick a representative value,