switch out deprecated extractor

tidymodels · Jul 22, 2021 · c8ba315 · c8ba315
1 parent 1259c51
commit c8ba315
Show file tree

Hide file tree

Showing 2 changed files with 3 additions and 3 deletions.
diff --git a/R/add_candidates.R b/R/add_candidates.R
@@ -184,7 +184,7 @@ add_candidates.default <- function(data_stack, candidates, name, ...) {
 .set_mode_ <- function(stack, candidates, name) {
   wf_spec <- 
     attr(candidates, "workflow") %>%
-    workflows::pull_workflow_spec()
+    workflows::extract_spec_parsnip()
 
   new_mode <- wf_spec$mode
   old_mode <- attr(stack, "mode")
@@ -363,7 +363,7 @@ update_stack_data <- function(stack, new_data) {
 stack_workflow <- function(x) {
   res <-
     workflows::workflow() %>%
-    workflows::add_model(workflows::pull_workflow_spec(x))
+    workflows::add_model(workflows::extract_spec_parsnip(x))
 
   pre <- workflows::pull_workflow_preprocessor(x)
 

diff --git a/R/print.R b/R/print.R
@@ -84,7 +84,7 @@ top_coefs <- function(x, penalty = x$penalty$penalty, n = 10) {
   sub_models <-
     purrr::map_dfr(x$cols_map, ~ tibble::tibble(terms = .x), .id = "model_name")
   model_types <- 
-    purrr::map(x$model_defs, workflows::pull_workflow_spec) %>% 
+    purrr::map(x$model_defs, workflows::extract_spec_parsnip) %>% 
     purrr::map_dfr(~ tibble::tibble(model_type = class(.x)[1]), .id = "model_name")
   res <- 
     dplyr::left_join(betas, sub_models, by = "terms") %>%