support tidymodels and multiple outputs in predict.cube [experimental]

appelmar · Jan 31, 2024 · cccbc94 · cccbc94
1 parent c6eb0bc
commit cccbc94
Showing 1 changed file with 1 addition and 1 deletion.
diff --git a/inst/scripts/f_predict.R b/inst/scripts/f_predict.R
@@ -20,7 +20,7 @@ out = array(NA, dim=c(length(output_names),nobs))
 if (is.list(p)) {
   for (i in 1:length(output_names)) {
     z = p[[output_names[i]]]
-    if (!is.numeric(z) && is.factor(z)) {
+    if (!is.numeric(z) && !is.factor(z)) {
       stop("Unexpected output from predict().")
     }
     out[i,idx] = as.numeric(z)