Update sarix.py to address missing values in input nhsn data

reichlab · Nov 19, 2024 · 6b95c61 · 6b95c61
1 parent a37e9f2
commit 6b95c61
Showing 1 changed file with 1 addition and 0 deletions.
diff --git a/src/idmodels/sarix.py b/src/idmodels/sarix.py
@@ -30,6 +30,7 @@ def run(self, run_config):
         df["xmas_spike"] = np.maximum(3 - np.abs(df["delta_xmas"]), 0)
 
         xy_colnames = ["inc_trans_cs"] + self.model_config.x
+        df = df.query("wk_end_date >= '2022-10-01'").interpolate()
         batched_xy = df[xy_colnames].values.reshape(len(df["location"].unique()), -1, len(xy_colnames))
 
         sarix_fit_all_locs_theta_pooled = sarix.SARIX(