Merge pull request #1175 from tidymodels/add-missing-drop-false

EmilHvitfeldt · web-flow · commit 474152f36990 · 2024-09-05T14:15:26.000-07:00
add missing drop = FALSE
diff --git a/R/predict.R b/R/predict.R
@@ -452,7 +452,7 @@ prepare_data <- function(object, new_data) {
   } else if (translate_from_xy_to_formula) {
     new_data <- .convert_xy_to_form_new(object$preproc, new_data)
   } else if (translate_from_xy_to_xy) {
-    new_data <- new_data[, object$preproc$x_names]
+    new_data <- new_data[, object$preproc$x_names, drop = FALSE]
   }
 
   encodings <- get_encoding(class(object$spec)[1])

Original file line number	Diff line number	Diff line change
`@@ -452,7 +452,7 @@ prepare_data <- function(object, new_data) {`
`452`	`452`	`} else if (translate_from_xy_to_formula) {`
`453`	`453`	`new_data <- .convert_xy_to_form_new(object$preproc, new_data)`
`454`	`454`	`} else if (translate_from_xy_to_xy) {`
`455`		`- new_data <- new_data[, object$preproc$x_names]`
	`455`	`+ new_data <- new_data[, object$preproc$x_names, drop = FALSE]`
`456`	`456`	`}`
`457`	`457`
`458`	`458`	`encodings <- get_encoding(class(object$spec)[1])`