address slowdown in form_form() (#1073)

tidymodels · Feb 26, 2024 · 5491946 · 5491946
1 parent 673b7d5
commit 5491946
Showing 1 changed file with 5 additions and 2 deletions.
diff --git a/R/fit_helpers.R b/R/fit_helpers.R
@@ -9,9 +9,12 @@ form_form <-
     if (inherits(env$data, "data.frame")) {
       check_outcome(eval_tidy(rlang::f_lhs(env$formula), env$data), object)
 
+      encoding_info <- get_encoding(class(object)[1])
       encoding_info <-
-        get_encoding(class(object)[1]) %>%
-        dplyr::filter(mode == object$mode, engine == object$engine)
+        vctrs::vec_slice(
+          encoding_info,
+          encoding_info$mode == object$mode & encoding_info$engine == object$engine
+        )
 
       remove_intercept <- encoding_info %>% dplyr::pull(remove_intercept)
       if (remove_intercept) {