update to latest iddata api

reichlab · Nov 26, 2024 · fc860f4 · fc860f4
1 parent a09605f
commit fc860f4
Show file tree

Hide file tree

Showing 4 changed files with 16 additions and 2 deletions.
diff --git a/requirements/requirements-dev.txt b/requirements/requirements-dev.txt
@@ -38,7 +38,7 @@ frozenlist==1.5.0
     #   aiosignal
 fsspec==2024.10.0
     # via s3fs
-iddata @ git+https://github.com/reichlab/iddata@b94caa9735d010059ea4117631c7b1908adff70d
+iddata @ git+https://github.com/reichlab/iddata@7d85f006345c8e17139588d24650509a1eef6af0
     # via idmodels (pyproject.toml)
 identify==2.6.1
     # via pre-commit

diff --git a/requirements/requirements.txt b/requirements/requirements.txt
@@ -30,7 +30,7 @@ frozenlist==1.5.0
     #   aiosignal
 fsspec==2024.10.0
     # via s3fs
-iddata @ git+https://github.com/reichlab/iddata@dddfeddefff101f7fafda11245333772aabef7b0
+iddata @ git+https://github.com/reichlab/iddata@7d85f006345c8e17139588d24650509a1eef6af0
     # via idmodels (pyproject.toml)
 idna==3.10
     # via yarl

diff --git a/tests/integration/test_gbqr.py b/tests/integration/test_gbqr.py
@@ -34,10 +34,17 @@ def test_gbqr(tmp_path):
 
 
     run_config = SimpleNamespace(
+        disease="flu",
         ref_date=datetime.date.fromisoformat("2024-01-06"),
         output_root=tmp_path / "model-output",
         artifact_store_root=tmp_path / "artifact-store",
         save_feat_importance=False,
+        locations=["US", "01", "02", "04", "05", "06", "08", "09", "10", "11",
+                   "12", "13", "15", "16", "17", "18", "19", "20", "21", "22",
+                   "23", "24", "25", "26", "27", "28", "29", "30", "31", "32",
+                   "33", "34", "35", "36", "37", "38", "39", "40", "41", "42",
+                   "44", "45", "46", "47", "48", "49", "50", "51", "53", "54",
+                   "55", "56", "72"],
         max_horizon=3,
         q_levels = [0.025, 0.50, 0.975],
         q_labels = ["0.025", "0.5", "0.975"],

diff --git a/tests/integration/test_sarix.py b/tests/integration/test_sarix.py
@@ -38,10 +38,17 @@ def test_sarix(tmp_path):
     )
 
     run_config = SimpleNamespace(
+        disease="flu",
         ref_date=datetime.date.fromisoformat("2024-01-06"),
         output_root=tmp_path / "model-output",
         artifact_store_root=tmp_path / "artifact-store",
         save_feat_importance=False,
+        locations=["US", "01", "02", "04", "05", "06", "08", "09", "10", "11",
+                   "12", "13", "15", "16", "17", "18", "19", "20", "21", "22",
+                   "23", "24", "25", "26", "27", "28", "29", "30", "31", "32",
+                   "33", "34", "35", "36", "37", "38", "39", "40", "41", "42",
+                   "44", "45", "46", "47", "48", "49", "50", "51", "53", "54",
+                   "55", "56", "72"],
         max_horizon=3,
         q_levels = [0.025, 0.50, 0.975],
         q_labels = ["0.025", "0.5", "0.975"],