Mutability

CDCgov · Feb 18, 2025 · 01f3781 · 01f3781
1 parent 373d3ce
commit 01f3781
Showing 1 changed file with 2 additions and 1 deletion.
diff --git a/pipelines/prep_data.py b/pipelines/prep_data.py
@@ -326,12 +326,13 @@ def get_pmfs(param_estimates: pl.LazyFrame, state_abb: str, disease: str):
         .collect(streaming=True)
         .get_column("value")
         .item(0)
-        .to_numpy()
+        .to_list()
     )
 
     # ensure 0 first entry; we do not model the possibility
     # of a zero infection-to-recorded-admission delay in Pyrenew-HEW
     delay_pmf[0] = 0
+    delay_pmf = np.array(delay_pmf)
     delay_pmf = delay_pmf / delay_pmf.sum()
 
     right_truncation_pmf = (