Add input validation

R/estimate_infections.R

@@ -455,26 +455,15 @@ fit_model_with_nuts <- function(args, future = FALSE, max_execution_time = Inf,
     }
   }
 
-  if (!future) {
-    fit <- fit_chain(1,
-      stan_args = args, max_time = max_execution_time,
-      catch = !id %in% c("estimate_infections", "epinow")
-    )
-    if (stuck_chains > 0) {
-      fit <- NULL
-    }
-    if (is.null(fit)) {
-      rlang::abort("model fitting was timed out or failed")
-    }
-  } else {
+  if (future) {
     chains <- args$chains
     args$chains <- 1
     args$cores <- 1
     fits <- future.apply::future_lapply(1:chains, fit_chain,
-      stan_args = args,
-      max_time = max_execution_time,
-      catch = TRUE,
-      future.seed = TRUE
+                                        stan_args = args,
+                                        max_time = max_execution_time,
+                                        catch = TRUE,
+                                        future.seed = TRUE
     )
     if (stuck_chains > 0) {
       fits[1:stuck_chains] <- NULL
@@ -498,12 +487,23 @@ fit_model_with_nuts <- function(args, future = FALSE, max_execution_time = Inf,
         if ((chains - failed_chains) < 2) {
           rlang::abort(
             "model fitting failed as too few chains were returned to assess",
-             " convergence (2 or more required)"
+            " convergence (2 or more required)"
           )
         }
       }
       fit <- rstan::sflist2stanfit(fit)
     }
+  } else {
+    fit <- fit_chain(1,
+        stan_args = args, max_time = max_execution_time,
+        catch = !id %in% c("estimate_infections", "epinow")
+    )
+    if (stuck_chains > 0) {
+      fit <- NULL
+    }
+    if (is.null(fit)) {
+      rlang::abort("model fitting was timed out or failed")
+    }
   }
   return(fit)
 }