Skip to content

Commit 4998f5b

Browse files
committed
better handling of data argument to formulas to avoid embedding data in calls
1 parent a4b3b86 commit 4998f5b

File tree

1 file changed

+6
-2
lines changed

1 file changed

+6
-2
lines changed

R/fitter.R

Lines changed: 6 additions & 2 deletions
Original file line numberDiff line numberDiff line change
@@ -153,9 +153,13 @@ formula_to_formula <-
153153
fit_args <- object$method$fit_args
154154

155155
if (isTRUE(unname(object$method$fit_name["pkg"] == "sparklyr"))) {
156-
fit_args$x <- data
156+
x <- data
157+
fit_args$x <- quote(x)
157158
} else {
158-
fit_args$data <- data
159+
if (is.name(data) | is.call(data))
160+
fit_args$data <- data
161+
else
162+
fit_args$data <- quote(data)
159163
}
160164
fit_args$formula <- formula
161165

0 commit comments

Comments
 (0)