address slowdown in fit_xy() (#1075)

tidymodels · Feb 27, 2024 · 60f97cd · 60f97cd
1 parent 5491946
commit 60f97cd
Showing 1 changed file with 7 additions and 4 deletions.
diff --git a/R/fit_helpers.R b/R/fit_helpers.R
@@ -175,11 +175,14 @@ xy_form <- function(object, env, control, ...) {
 
   check_outcome(env$y, object)
 
-  encoding_info <-
-    get_encoding(class(object)[1]) %>%
-    dplyr::filter(mode == object$mode, engine == object$engine)
+  encoding_info <- get_encoding(class(object)[1])
+  encoding_info <- 
+    vctrs::vec_slice(
+      encoding_info, 
+      encoding_info$mode == object$mode & encoding_info$engine == object$engine
+    )
 
-  remove_intercept <- encoding_info %>% dplyr::pull(remove_intercept)
+  remove_intercept <- encoding_info$remove_intercept
 
   data_obj <-
     .convert_xy_to_form_fit(