access attribute correctly

CDCgov · Jan 18, 2025 · b672c39 · b672c39
1 parent 5eb688c
commit b672c39
Showing 1 changed file with 2 additions and 2 deletions.
diff --git a/pyrenew_hew/pyrenew_hew_model.py b/pyrenew_hew/pyrenew_hew_model.py
@@ -504,12 +504,12 @@ def sample(
         sample_hospital_admissions: bool = False,
         sample_wastewater: bool = False,
     ) -> dict[str, ArrayLike]:  # numpydoc ignore=GL08
+        n_init_days = self.latent_infection_process_rv.n_initialization_points
         latent_infections = self.latent_infection_process_rv(
             n_days_post_init=data.n_days_post_init,
         )
         first_latent_infection_dow = (
-            data.first_data_date_overall
-            - datetime.timedelta(days=self.n_initialization_points)
+            data.first_data_date_overall - datetime.timedelta(days=n_init_days)
         ).weekday()
 
         observed_ed_visits = None