Replace cumsum broadcast with accumulate

CDCgov · Oct 4, 2024 · 3f68792 · 3f68792
1 parent 21fb991
commit 3f68792
Showing 1 changed file with 1 addition and 1 deletion.
diff --git a/EpiAware/src/EpiInfModels/ExpGrowthRate.jl b/EpiAware/src/EpiInfModels/ExpGrowthRate.jl
@@ -116,5 +116,5 @@ I_t = generated_quantities(latent_inf, θ)
 "
 @model function EpiAwareBase.generate_latent_infs(epi_model::ExpGrowthRate, rt)
     init_incidence ~ epi_model.initialisation_prior
-    return exp.(init_incidence .+ cumsum(rt))
+    return exp.(accumulate(+, rt; init=init_incidence))
 end