don't try this for spark

tidymodels · Dec 8, 2023 · 2b85412 · 2b85412
1 parent 0372970
commit 2b85412
Showing 1 changed file with 9 additions and 9 deletions.
diff --git a/R/fit_helpers.R b/R/fit_helpers.R
@@ -6,17 +6,17 @@
 form_form <-
   function(object, control, env, ...) {
 
-    encoding_info <-
-      get_encoding(class(object)[1]) %>%
-      dplyr::filter(mode == object$mode, engine == object$engine)
-
-    remove_intercept <- encoding_info %>% dplyr::pull(remove_intercept)
-    if (remove_intercept) {
-      env$data <- env$data[, colnames(env$data) != "(Intercept)", drop = FALSE]
-    }
-
     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]) %>%
+        dplyr::filter(mode == object$mode, engine == object$engine)
+
+      remove_intercept <- encoding_info %>% dplyr::pull(remove_intercept)
+      if (remove_intercept) {
+        env$data <- env$data[, colnames(env$data) != "(Intercept)", drop = FALSE]
+      }
     }
 
     # prob rewrite this as simple subset/levels