actually use pathfinder

epiforecasts · Mar 22, 2024 · 3e24291 · 3e24291
1 parent f7055f3
commit 3e24291
Showing 1 changed file with 3 additions and 1 deletion.
diff --git a/R/estimate_infections.R b/R/estimate_infections.R
@@ -547,8 +547,10 @@ fit_model_approximate <- function(args, future = FALSE, id = "stan") {
     } else if (inherits(stan_args$object, "CmdStanModel")) {
       if (method == "vb") {
         sample_func <- stan_args$object$variational
-      } else {
+      } else if (method == "laplace") {
         sample_func <- stan_args$object$laplace
+      } else {
+        sample_func <- stan_args$object$pathfinder
       }
       stan_args$object <- NULL
     }