fixing logic

cmu-delphi · Sep 13, 2024 · 2654946 · 2654946
1 parent f25605d
commit 2654946
Show file tree

Hide file tree

Showing 2 changed files with 2 additions and 5 deletions.
diff --git a/google_symptoms/delphi_google_symptoms/date_utils.py b/google_symptoms/delphi_google_symptoms/date_utils.py
@@ -4,7 +4,7 @@
 from itertools import product
 from typing import Dict, List, Union
 
-import covidcast
+from delphi_epidata import Epidata
 from delphi_utils.validator.utils import lag_converter
 from pandas import to_datetime
 
@@ -76,7 +76,7 @@ def generate_num_export_days(params: Dict, logger) -> [int]:
     num_export_days = params["indicator"]["num_export_days"]
     custom_run = False if not params["common"].get("custom_run") else params["common"].get("custom_run", False)
 
-    if num_export_days is None:
+    if num_export_days is None and not custom_run:
         # Generate a list of signals we expect to produce
         sensor_names = set(
             "_".join([metric, smoother, "search"])

diff --git a/google_symptoms/delphi_google_symptoms/run.py b/google_symptoms/delphi_google_symptoms/run.py
@@ -9,10 +9,7 @@
 from itertools import product
 
 import numpy as np
-from delphi_epidata import Epidata
 from delphi_utils import create_export_csv, get_structured_logger
-from delphi_utils.validator.utils import lag_converter
-from pandas import to_datetime
 
 from .constants import COMBINED_METRIC, FULL_BKFILL_START_DATE, GEO_RESOLUTIONS, SMOOTHERS, SMOOTHERS_MAP
 from .date_utils import generate_num_export_days