actually use s3 predict...

tidymodels · Jul 8, 2024 · 12c064a · 12c064a
1 parent d225cb0
commit 12c064a
Show file tree

Hide file tree

Showing 3 changed files with 5 additions and 5 deletions.
diff --git a/NAMESPACE b/NAMESPACE
@@ -16,12 +16,12 @@ S3method(orbital,step_scale)
 S3method(orbital,step_select)
 S3method(orbital,step_zv)
 S3method(orbital,workflow)
+S3method(predict,orbital_class)
 S3method(print,orbital_class)
 export(orbital)
 export(orbital_dt)
 export(orbital_inline)
 export(orbital_json_read)
 export(orbital_json_write)
 export(orbital_sql)
-export(predict)
 export(to_r_fun)
diff --git a/R/predict.R b/R/predict.R
@@ -24,7 +24,7 @@
 #'
 #' predict(orbital_obj, mtcars)
 #' @export
-predict <- function(object, new_data, ...) {
+predict.orbital_class <- function(object, new_data, ...) {
   rlang::check_dots_empty()
   res <- dplyr::mutate(new_data, !!!orbital_inline(object))
 

diff --git a/man/predict.Rd → man/predict.orbital_class.Rd b/man/predict.Rd → man/predict.orbital_class.Rd