better support for non-standard x/y interface arg names

tidymodels · Apr 7, 2020 · 0ddb2cd · 0ddb2cd
1 parent 6e8323c
commit 0ddb2cd
Showing 1 changed file with 2 additions and 2 deletions.
diff --git a/R/fit_helpers.R b/R/fit_helpers.R
@@ -89,8 +89,8 @@ xy_xy <- function(object, env, control, target = "none", ...) {
   # sub in arguments to actual syntax for corresponding engine
   object <- translate(object, engine = object$engine)
 
-  object$method$fit$args[["y"]] <- quote(y)
-  object$method$fit$args[["x"]] <-
+  object$method$fit$args[[ object$method$fit$protect[2]] ] <- quote(y)
+  object$method$fit$args[[ object$method$fit$protect[1]] ] <-
     switch(
       target,
       none = quote(x),