support covid

reichlab · Nov 20, 2024 · c87a101 · c87a101
1 parent 05a2f98
commit c87a101
Showing 1 changed file with 1 addition and 1 deletion.
diff --git a/src/idmodels/gbqr.py b/src/idmodels/gbqr.py
@@ -32,7 +32,7 @@ def run(self, run_config):
             flusurvnet_kwargs = {"burden_adj": False}
 
         fdl = FluDataLoader()
-        df = fdl.load_data(nhsn_kwargs={"as_of": run_config.ref_date},
+        df = fdl.load_data(nhsn_kwargs={"as_of": run_config.ref_date, disease=run_config.disease},
                            ilinet_kwargs=ilinet_kwargs,
                            flusurvnet_kwargs=flusurvnet_kwargs,
                            sources=self.model_config.sources,