add initial_checked_models to ptc options models option

hubverse-org · Nov 21, 2024 · 6aee4ee · 6aee4ee
1 parent 41763e3
commit 6aee4ee
Show file tree

Hide file tree

Showing 3 changed files with 5 additions and 4 deletions.
diff --git a/src/hub_predtimechart/generate_options.py b/src/hub_predtimechart/generate_options.py
@@ -56,7 +56,8 @@ def get_max_ref_date_or_first_config_ref_date(reference_dates):
     # set `models` and `initial_checked_models`
     options['models'] = []
     for model_id, metadata in hub_config.model_id_to_metadata.items():
-        options['models'].append(model_id)
+        if metadata['designated_model'] or model_id in hub_config.initial_checked_models:
+            options['models'].append(model_id)
     options['models'].sort()
     options['initial_checked_models'] = hub_config.initial_checked_models
 

diff --git a/tests/expected/FluSight-forecast-hub/predtimechart-options.json b/tests/expected/FluSight-forecast-hub/predtimechart-options.json
@@ -72,9 +72,9 @@
   "initial_as_of": "2023-10-07",
   "current_date": "2023-10-07",
   "models": [
-    "CADPH-FluCAT_Ensemble", "CEPH-Rtrend_fluH", "CMU-TimeSeries", "CU-ensemble", "GT-FluFNP", "ISU_NiemiLab-NLH", "JHU_CSSE-CSSE_Ensemble", "LUcompUncertLab-chimera", "LosAlamos_NAU-CModel_Flu", "MIGHTE-Nsemble", "MOBS-GLEAM_FLUH", "NIH-Flu_ARIMA", "NU_UCSD-GLEAM_AI_FLUH", "PSI-PROF", "SGroup-RandomForest", "SigSci-CREG", "SigSci-TSENS", "Stevens-GBR", "UGA_flucast-Copycat", "UGA_flucast-INFLAenza", "UGuelph-CompositeCurve", "UGuelphensemble-GRYPHON", "UM-DeepOutbreak", "UMass-flusion", "UMass-trends_ensemble", "UNC_IDD-InfluPaint", "UVAFluX-Ensemble", "VTSanghani-Ensemble", "cfa-flumech", "cfarenewal-cfaepimlight", "fjordhest-ensemble"
+    "CADPH-FluCAT_Ensemble", "CEPH-Rtrend_fluH", "CMU-TimeSeries", "CU-ensemble", "FluSight-baseline", "FluSight-ensemble", "GT-FluFNP", "ISU_NiemiLab-NLH", "JHU_CSSE-CSSE_Ensemble", "LUcompUncertLab-chimera", "LosAlamos_NAU-CModel_Flu", "MIGHTE-Nsemble", "MOBS-GLEAM_FLUH", "NIH-Flu_ARIMA", "NU_UCSD-GLEAM_AI_FLUH", "PSI-PROF", "SGroup-RandomForest", "SigSci-CREG", "SigSci-TSENS", "Stevens-GBR", "UGA_flucast-Copycat", "UGA_flucast-INFLAenza", "UGuelph-CompositeCurve", "UGuelphensemble-GRYPHON", "UM-DeepOutbreak", "UMass-flusion", "UMass-trends_ensemble", "UNC_IDD-InfluPaint", "UVAFluX-Ensemble", "VTSanghani-Ensemble", "cfa-flumech", "cfarenewal-cfaepimlight", "fjordhest-ensemble"
   ],
-  "initial_checked_models": ["UMass-flusion", "UMass-trends_ensemble"],
+  "initial_checked_models": ["FluSight-baseline", "FluSight-ensemble"],
   "disclaimer": "Most forecasts have failed to reliably predict rapid changes in the trends of reported cases and hospitalizations. Due to this limitation, they should not be relied upon for decisions about the possibility or timing of rapid changes in trends.",
   "initial_xaxis_range": null
 }
diff --git a/tests/hubs/FluSight-forecast-hub/hub-config/predtimechart-config.yml b/tests/hubs/FluSight-forecast-hub/hub-config/predtimechart-config.yml
@@ -4,5 +4,5 @@ model_tasks_idx: 1
 reference_date_col_name: 'reference_date'
 target_date_col_name: 'target_end_date'
 horizon_col_name: 'horizon'
-initial_checked_models: ['UMass-flusion', 'UMass-trends_ensemble']
+initial_checked_models: ['FluSight-baseline', 'FluSight-ensemble']
 disclaimer: Most forecasts have failed to reliably predict rapid changes in the trends of reported cases and hospitalizations. Due to this limitation, they should not be relied upon for decisions about the possibility or timing of rapid changes in trends.