LatentDelay passes a view of the obs into comparison with expected_obs

CDCgov · Jul 22, 2024 · 95af1de · 95af1de
1 parent cc8e0d2
commit 95af1de
Showing 1 changed file with 4 additions and 3 deletions.
diff --git a/EpiAware/src/EpiObsModels/modifiers/LatentDelay.jl b/EpiAware/src/EpiObsModels/modifiers/LatentDelay.jl
@@ -72,10 +72,11 @@ Generates observations based on the `LatentDelay` observation model.
         vcat(trunc_Y_t[(pmf_length + 1):end], 0.0)
     )
 
-    complete_obs = vcat(fill(missing, pmf_length + first_Y_t - 2), expected_obs)
-
+    # complete_obs = vcat(fill(missing, pmf_length + first_Y_t - 2), expected_obs)
+    m = size(expected_obs, 1)
+    _y_t = ismissing(y_t) ? missing : @view y_t[(end - m + 1):end]
     @submodel y_t = generate_observations(
-        obs_model.model, y_t, complete_obs)
+        obs_model.model, _y_t, expected_obs)
 
     return y_t
 end