Merge pull request #121 from malcolmbarrett/tidy_model_object

topepo · web-flow · commit 52ddb78488de · 2019-02-28T20:41:10.000-05:00
add tidy() method for model_fit objects
diff --git a/DESCRIPTION b/DESCRIPTION
@@ -18,7 +18,7 @@ Depends:
     R (>= 2.10)
 Imports: 
     dplyr,
-    rlang (>=  0.3.0.1),
+    rlang (>= 0.3.1),
     purrr,
     utils,
     tibble,
diff --git a/NAMESPACE b/NAMESPACE
@@ -106,6 +106,7 @@ export(show_call)
 export(surv_reg)
 export(svm_poly)
 export(svm_rbf)
+export(tidy.model_fit)
 export(translate)
 export(varying)
 export(varying_args)
diff --git a/R/tidy.R b/R/tidy.R
@@ -0,0 +1,9 @@
+#' Turn a parsnip model object into a tidy tibble
+#'
+#' This method tidies the model in a parsnip model object, if it exists.
+#'
+#' @inheritParams generics::tidy
+#'
+#' @return a tibble
+#' @export
+tidy.model_fit <- function(x, ...) generics::tidy(x$fit, ...)
diff --git a/man/tidy.model_fit.Rd b/man/tidy.model_fit.Rd
