Skip to content

Commit

Permalink
generalized target data generation to both flu and covid
Browse files Browse the repository at this point in the history
  • Loading branch information
matthewcornell committed Dec 3, 2024
1 parent c099ad1 commit 758c92b
Showing 1 changed file with 2 additions and 1 deletion.
3 changes: 2 additions & 1 deletion src/hub_predtimechart/app/generate_target_json_files.py
Original file line number Diff line number Diff line change
@@ -1,4 +1,5 @@
import json
import sys
from datetime import date, timedelta
from pathlib import Path

Expand Down Expand Up @@ -53,7 +54,7 @@ def main(hub_dir, ptc_config_file, target_out_dir):
target_data_df = get_target_data_df(hub_dir, hub_config.target_data_file_name)
except FileNotFoundError as error:
logger.error(f"target data file not found. {hub_config.target_data_file_name=}, {error=}")
return
sys.exit(1)

for loc in target_data_df['location'].unique():
task_ids_tuple = (loc,)
Expand Down

0 comments on commit 758c92b

Please sign in to comment.