add correct package to fit function

microsoft · Apr 10, 2024 · 70718e0 · 70718e0
1 parent ce26c0c
commit 70718e0
Show file tree

Hide file tree

Showing 2 changed files with 2 additions and 2 deletions.
diff --git a/R/multistep_glmnet.R b/R/multistep_glmnet.R
@@ -335,7 +335,7 @@ glmnet_multistep_fit_impl <- function(x, y,
 
     # fit model
     fit_glmnet <- linreg_reg_spec %>%
-      fit(Target ~ ., data = combined_df)
+      generics::fit(Target ~ ., data = combined_df)
 
     # create prediction
     glmnet_fitted <- predict(fit_glmnet, combined_df)

diff --git a/R/multistep_mars.R b/R/multistep_mars.R
@@ -358,7 +358,7 @@ mars_multistep_fit_impl <- function(x, y,
 
     # fit model
     fit_mars <- mars_reg_spec %>%
-      fit(Target ~ ., data = combined_df)
+      generics::fit(Target ~ ., data = combined_df)
 
     # create prediction
     mars_fitted <- predict(fit_mars, combined_df)