From c87a1014bec61d67fc84da1c916ccadb2e746b77 Mon Sep 17 00:00:00 2001
From: Evan Ray <elray@umass.edu>
Date: Wed, 20 Nov 2024 16:22:25 -0500
Subject: [PATCH] support covid

---
 src/idmodels/gbqr.py | 2 +-
 1 file changed, 1 insertion(+), 1 deletion(-)

diff --git a/src/idmodels/gbqr.py b/src/idmodels/gbqr.py
index b838d33..10c80cf 100644
--- 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,