From 2654946113a538e81fb6a636b5385b65d2d05534 Mon Sep 17 00:00:00 2001 From: Amaris Sim Date: Fri, 13 Sep 2024 11:48:14 -0400 Subject: [PATCH] fixing logic --- google_symptoms/delphi_google_symptoms/date_utils.py | 4 ++-- google_symptoms/delphi_google_symptoms/run.py | 3 --- 2 files changed, 2 insertions(+), 5 deletions(-) diff --git a/google_symptoms/delphi_google_symptoms/date_utils.py b/google_symptoms/delphi_google_symptoms/date_utils.py index fc3ed5bd2..b32c4f3f8 100644 --- 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 index 3e7b8db05..8303a9a8a 100644 --- 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