Update model/src/pyrenew/latent/infectionswithfeedback.py

CDCgov · Jul 25, 2024 · ee817e5 · ee817e5
1 parent 6026b19
commit ee817e5
Showing 1 changed file with 3 additions and 1 deletion.
diff --git a/model/src/pyrenew/latent/infectionswithfeedback.py b/model/src/pyrenew/latent/infectionswithfeedback.py
@@ -200,7 +200,9 @@ def sample(
 
         return InfectionsRtFeedbackSample(
             post_initialization_infections=SampledValue(
-                post_initialization_infections
+                value=post_initialization_infections,
+                t_start=self.t_start,
+                t_unit=self.t_unit)
             ),
             rt=SampledValue(Rt_adj, t_start=self.t_start, t_unit=self.t_unit),
         )