backend-specific success criteria

epiforecasts · Feb 12, 2024 · e286ffe · e286ffe
1 parent 7953226
commit e286ffe
Showing 1 changed file with 2 additions and 1 deletion.
diff --git a/R/estimate_infections.R b/R/estimate_infections.R
@@ -450,7 +450,8 @@ fit_model_with_nuts <- function(args, future = FALSE, max_execution_time = Inf,
       )
     }
 
-    if (inherits(fit, c("stanfit", "CmdStanMCMC"))) {
+    if ((inherits(fit, "stanfit") && fit@mode != 2L) ||
+        inherits(fit, "CmdStanMCMC")) {
       return(fit)
     } else {
       return(NULL)