diff --git a/sir_complainsalot/delphi_sir_complainsalot/check_source.py b/sir_complainsalot/delphi_sir_complainsalot/check_source.py index 840575cbf..380142301 100644 --- a/sir_complainsalot/delphi_sir_complainsalot/check_source.py +++ b/sir_complainsalot/delphi_sir_complainsalot/check_source.py @@ -8,6 +8,7 @@ import pandas as pd covidcast.covidcast._ASYNC_CALL = True # pylint: disable=protected-access +covidcast.covidcast.VALID_GEO_TYPES.add("hsa_nci") # workaround to use new geo_type w/ legacy client @dataclass class Complaint: @@ -72,7 +73,7 @@ def check_source(data_source, meta, params, grace, logger): # pylint: disable=t end_day=datetime.now().strftime("%Y-%m-%d"), geo_type=row["geo_type"]) - latest_data = covidcast.signal( + latest_data = covidcast.covidcast.signal( data_source, row["signal"], start_day=datetime.now() - timedelta(days = 14), end_day=datetime.now(),