similar modifications for numeric predictions

Author: topepo

R/predict.R

@@ -185,7 +185,7 @@ format_num <- function(x) {
   if (inherits(x, "tbl_spark"))
     return(x)
 
-  if (isTRUE(ncol(x) > 1)) {
+  if (isTRUE(ncol(x) > 1) | is.data.frame(x)) {
     x <- as_tibble(x, .name_repair = "minimal")
     if (!any(grepl("^\\.pred", names(x)))) {
       names(x) <- paste0(".pred_", names(x))

R/predict_class.R

@@ -42,7 +42,7 @@ predict_class.model_fit <- function(object, new_data, ...) {
   } else {
     if (!inherits(res, "tbl_spark")) {
       # Now case where a parsnip model generated `res`
-      if (tibble::is_tibble(res) && ncol(res) == 1 && is.factor(res[[1]])) {
+      if (is.data.frame(res) && ncol(res) == 1 && is.factor(res[[1]])) {
         res <- res[[1]]
       } else {
         res$values <- factor(as.character(res$values), levels = object$lvl)