make the removal work with spark

topepo · topepo · commit bb5e98e17202 · 2020-07-30T11:05:18.000-04:00
diff --git a/R/predict.R b/R/predict.R
@@ -245,7 +245,7 @@ prepare_data <- function(object, new_data) {
     dplyr::filter(mode == object$spec$mode, engine == object$spec$engine) %>%
     dplyr::pull(remove_intercept)
   if (remove_intercept) {
-    new_data <- new_data[, colnames(new_data) != "(Intercept)", drop = FALSE]
+    new_data <- new_data %>% dplyr::select(-dplyr::one_of("(Intercept)"))
   }
 
   switch(

Original file line number	Diff line number	Diff line change
`@@ -245,7 +245,7 @@ prepare_data <- function(object, new_data) {`
`245`	`245`	`dplyr::filter(mode == object$spec$mode, engine == object$spec$engine) %>%`
`246`	`246`	`dplyr::pull(remove_intercept)`
`247`	`247`	`if (remove_intercept) {`
`248`		`- new_data <- new_data[, colnames(new_data) != "(Intercept)", drop = FALSE]`
	`248`	`+ new_data <- new_data %>% dplyr::select(-dplyr::one_of("(Intercept)"))`
`249`	`249`	`}`
`250`	`250`
`251`	`251`	`switch(`