don't use future_lapply for cmdstanr

epiforecasts · Feb 4, 2024 · 8c789c4 · 8c789c4
1 parent ced14ad
commit 8c789c4
Showing 1 changed file with 8 additions and 3 deletions.
diff --git a/R/simulate_infections.R b/R/simulate_infections.R
@@ -245,12 +245,18 @@ simulate_infections <- function(estimates,
 
   safe_batch <- safely(batch_simulate)
 
+  if (backend == "cmdstanr") {
+    lapply_func <- lapply ## future_lapply can't handle cmdstanr
+  } else {
+    lapply_func <- function(...) future_lapply(future.seed = TRUE, ...)
+  }
+
   ## simulate in batches
   with_progress({
     if (verbose) {
       p <- progressor(along = batches)
     }
-    out <- future_lapply(batches,
+    out <- lapply_func(batches,
       function(batch) {
         if (verbose) {
           p()
@@ -260,8 +266,7 @@ simulate_infections <- function(estimates,
           shift, dates, batch[[1]],
           batch[[2]]
         )[[1]]
-      },
-      future.seed = TRUE
+      }
     )
   })