fix: change to extract_central_estimates to deal with 1-time estimates

epiverse-trace · Aug 1, 2024 · 962df9e · 962df9e
1 parent ea6093f
commit 962df9e
Showing 1 changed file with 2 additions and 1 deletion.
diff --git a/R/plot_seromodel.R b/R/plot_seromodel.R
@@ -107,7 +107,8 @@ extract_central_estimates <- function(
   alpha = 0.05,
   par_name = "foi_vector"
 ) {
-  samples <- rstan::extract(seromodel, par_name)[[1]]
+  samples <- rstan::extract(seromodel, par_name)[[1]] %>%
+    as.matrix() #to deal with 1-time estimates
   central_estimates <- data.frame(
     median = apply(samples, 2, quantile, 0.5),
     lower = apply(samples, 2, quantile, alpha),