From 01f37811d40b8482cfb29630d805baf4ac3c6442 Mon Sep 17 00:00:00 2001
From: "Dylan H. Morris" <dzl1@cdc.gov>
Date: Tue, 18 Feb 2025 13:52:46 -0500
Subject: [PATCH] Mutability

---
 pipelines/prep_data.py | 3 ++-
 1 file changed, 2 insertions(+), 1 deletion(-)

diff --git a/pipelines/prep_data.py b/pipelines/prep_data.py
index 47217377..551cccdb 100644
--- 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 = (