Merge pull request #14 from tidymodels/last_fit

tidymodels · Jul 11, 2024 · 0b98f8c · 0b98f8c
2 parents ba21ba4 + b77bb5b
commit 0b98f8c
Show file tree

Hide file tree

Showing 4 changed files with 10 additions and 0 deletions.
diff --git a/DESCRIPTION b/DESCRIPTION
@@ -20,6 +20,7 @@ Suggests:
     dbplyr,
     dplyr,
     dtplyr,
+    hardhat,
     jsonlite,
     kknn,
     parsnip,

diff --git a/NAMESPACE b/NAMESPACE
@@ -1,6 +1,7 @@
 # Generated by roxygen2: do not edit by hand
 
 S3method(orbital,default)
+S3method(orbital,last_fit)
 S3method(orbital,model_fit)
 S3method(orbital,model_spec)
 S3method(orbital,recipe)

diff --git a/NEWS.md b/NEWS.md
@@ -1,5 +1,7 @@
 # orbital (development version)
 
+* `orbital()` now works on `tune::last_fit()` objects. (#13)
+
 * `orbital_predict()` has been removed and replaced with the more idiomatic `predict()` method. (#10)
 
 # orbital 0.1.0

diff --git a/R/last_fit.R b/R/last_fit.R
@@ -0,0 +1,6 @@
+
+#' @export
+orbital.last_fit <- function(x, ...) {
+  x <- hardhat::extract_workflow(x)
+  orbital(x)
+}