From a09605fedb9cb41729053b785688a8d1e6a03544 Mon Sep 17 00:00:00 2001 From: Evan Ray Date: Wed, 20 Nov 2024 19:49:51 -0500 Subject: [PATCH] support covid for sarix --- src/idmodels/sarix.py | 6 +++--- 1 file changed, 3 insertions(+), 3 deletions(-) diff --git a/src/idmodels/sarix.py b/src/idmodels/sarix.py index 8775791..3fd904d 100644 --- a/src/idmodels/sarix.py +++ b/src/idmodels/sarix.py @@ -1,7 +1,7 @@ import numpy as np import pandas as pd -from iddata.loader import FluDataLoader +from iddata.loader import DiseaseDataLoader from iddata.utils import get_holidays from sarix import sarix @@ -13,8 +13,8 @@ def __init__(self, model_config): self.model_config = model_config def run(self, run_config): - fdl = FluDataLoader() - df = fdl.load_data(nhsn_kwargs={"as_of": run_config.ref_date}, + fdl = DiseaseDataLoader() + df = fdl.load_data(nhsn_kwargs={"as_of": run_config.ref_date, "disease": run_config.disease}, sources=self.model_config.sources, power_transform=self.model_config.power_transform) if run_config.locations is not None: