add a benchmark

CDCgov · Jul 24, 2024 · ee3e70e · ee3e70e
1 parent c97090b
commit ee3e70e
Show file tree

Hide file tree

Showing 2 changed files with 7 additions and 1 deletion.
diff --git a/EpiAware/src/EpiObsModels/modifiers/Aggregate.jl b/EpiAware/src/EpiObsModels/modifiers/Aggregate.jl
@@ -42,7 +42,7 @@ end
 
 @model function EpiAwareBase.generate_observations(ag::Aggregate, y_t, Y_t)
     if ismissing(y_t)
-        y_t = fill(0.0, length(Y_t))
+        y_t = Vector{Missing}(missing, length(Y_t))
     end
 
     n = length(y_t)

diff --git a/benchmark/bench/EpiObsModels/modifiers/Aggregate.jl b/benchmark/bench/EpiObsModels/modifiers/Aggregate.jl
@@ -0,0 +1,6 @@
+let
+    I_t = fill(10, 100)
+    weekly_agg = Aggregate(PoissonError(), [0, 0, 0, 0, 7, 0, 0])
+    mdl = generate_observations(weekly_agg, I_t, I_t)
+    suite["Aggregate"] = make_epiaware_suite(mdl)
+end