transition from deprecated map_df*() variants (#942)

Author: simonpcouch

NAMESPACE

@@ -359,8 +359,6 @@ importFrom(purrr,imap_lgl)
 importFrom(purrr,map)
 importFrom(purrr,map_chr)
 importFrom(purrr,map_dbl)
-importFrom(purrr,map_df)
-importFrom(purrr,map_dfr)
 importFrom(purrr,map_lgl)
 importFrom(rlang,abort)
 importFrom(rlang,call2)

R/aaa_models.R

@@ -957,10 +957,11 @@ show_model_info <- function(model) {
     cat(" engines: \n")
 
     weight_info <-
-      purrr::map_df(
+      purrr::map(
         model,
         ~ get_from_env(paste0(.x, "_fit")) %>% mutate(model = .x)
       ) %>%
+      purrr::list_rbind() %>%
       dplyr::mutate(protect = map(value, ~ .x$protect)) %>%
       dplyr::select(-value) %>%
       dplyr::mutate(

R/boost_tree.R

@@ -492,8 +492,10 @@ multi_predict._xgb.Booster <-
         type <- "numeric"
     }
 
-    res <- map_df(trees, xgb_by_tree, object = object, new_data = new_data,
-                  type = type, ...)
+    res <-
+      map(trees, xgb_by_tree, object = object, new_data = new_data,
+          type = type, ...) %>%
+      purrr::list_rbind()
     res <- arrange(res, .row, trees)
     res <- split(res[, -1], res$.row)
     names(res) <- NULL
@@ -612,8 +614,9 @@ multi_predict._C5.0 <-
       type <- "class"
 
     res <-
-      map_df(trees, C50_by_tree, object = object,
-             new_data = new_data, type = type, ...)
+      map(trees, C50_by_tree, object = object,
+          new_data = new_data, type = type, ...) %>%
+      purrr::list_rbind()
     res <- arrange(res, .row, trees)
     res <- split(res[, -1], res$.row)
     names(res) <- NULL

R/engine_docs.R

@@ -77,14 +77,16 @@ extensions <- function() {
 update_model_info_file <- function(path = "inst/models.tsv") {
   mods <- get_from_env("models")
   info <-
-    purrr::map_dfr(mods, ~ get_from_env(.x) %>% dplyr::mutate(model = .x)) %>%
+    purrr::map(mods, ~ get_from_env(.x) %>% dplyr::mutate(model = .x)) %>%
+    purrr::list_rbind() %>%
     dplyr::arrange(model, mode, engine) %>%
     dplyr::select(model, mode, engine)
   exts <-
-    purrr::map_dfr(
+    purrr::map(
       mods,
       ~ get_from_env(paste0(.x, "_pkgs")) %>% dplyr::mutate(model = .x)
     ) %>%
+    purrr::list_rbind() %>%
     tidyr::unnest(cols = "pkg") %>%
     dplyr::inner_join(tibble::tibble(pkg = extensions()), by = "pkg")
 
@@ -310,5 +312,5 @@ list_md_problems <- function() {
     tibble(basename(file), line, problem)
   }
 
-  purrr::map_dfr(md_files, get_errors)
+  purrr::map(md_files, get_errors) %>% purrr::list_rbind()
 }

R/install_packages.R

@@ -14,10 +14,11 @@ install_engine_packages <- function(extension = TRUE, extras = TRUE,
     bio_pkgs <- c(bio_pkgs, "mixOmics")
   }
 
-  engine_packages <- purrr::map_dfr(
+  engine_packages <- purrr::map(
     ls(envir = get_model_env(), pattern = "_pkgs$"),
     get_from_env
   ) %>%
+    purrr::list_rbind() %>%
     dplyr::pull(pkg) %>%
     unlist() %>%
     unique() %>%

R/mars.R

@@ -126,7 +126,8 @@ check_args.mars <- function(object) {
 
 earth_submodel_pred <- function(object, new_data, terms = 2:3, ...) {
   load_libs(object, quiet = TRUE, attach = TRUE)
-  map_dfr(terms, earth_reg_updater, object = object, newdata = new_data, ...)
+  map(terms, earth_reg_updater, object = object, newdata = new_data, ...) %>%
+    purrr::list_rbind()
 }
 
 earth_reg_updater <- function(num, object, new_data, ...) {
@@ -184,8 +185,9 @@ multi_predict._earth <-
     }
 
     res <-
-      map_df(num_terms, earth_by_terms, object = object,
-             new_data = new_data, type = type, ...)
+      map(num_terms, earth_by_terms, object = object,
+          new_data = new_data, type = type, ...) %>%
+      purrr::list_rbind()
     res <- arrange(res, .row, num_terms)
     res <- split(res[, -1], res$.row)
     names(res) <- NULL

R/mlp.R

@@ -359,7 +359,7 @@ mlp_num_weights <- function(p, hidden_units, classes) {
 
 ## -----------------------------------------------------------------------------
 
-#' @importFrom purrr map_df map
+#' @importFrom purrr map
 #' @importFrom dplyr arrange select
 #' @rdname multi_predict
 #' @param epochs An integer vector for the number of training epochs.
@@ -384,7 +384,8 @@ multi_predict._torch_mlp <-
       purrr::map(epochs,
                  ~ predict(object, new_data, type, epochs = .x) %>%
                    dplyr::mutate(epochs = .x)) %>%
-      purrr::map_dfr(~ .x %>% dplyr::mutate(.row = 1:nrow(new_data))) %>%
+      purrr::map(~ .x %>% dplyr::mutate(.row = 1:nrow(new_data))) %>%
+      purrr::list_rbind() %>%
       dplyr::arrange(.row, epochs)
     res <- split(dplyr::select(res, -.row), res$.row)
     names(res) <- NULL

R/nearest_neighbor.R

@@ -161,8 +161,9 @@ multi_predict._train.kknn <-
     }
 
     res <-
-      purrr::map_df(neighbors, knn_by_k, object = object,
-                    new_data = new_data, type = type, ...)
+      purrr::map(neighbors, knn_by_k, object = object,
+                 new_data = new_data, type = type, ...) %>%
+      purrr::list_rbind()
     res <- dplyr::arrange(res, .row, neighbors)
     res <- split(res[, -1], res$.row)
     names(res) <- NULL

R/parsnip-package.R

@@ -15,7 +15,7 @@
 #' @importFrom glue glue_collapse
 #' @importFrom lifecycle deprecated
 #' @importFrom pillar type_sum
-#' @importFrom purrr as_vector imap imap_lgl map map_chr map_dbl map_df map_dfr
+#' @importFrom purrr as_vector imap imap_lgl map map_chr map_dbl
 #' @importFrom purrr map_lgl %||%
 #' @importFrom rlang abort call2 caller_env current_env enquo enquos eval_tidy
 #' @importFrom rlang expr get_expr is_empty is_missing is_null is_quosure

R/varying.R

@@ -112,7 +112,7 @@ varying_args.recipe <- function(object, full = TRUE, ...) {
     return(varying_tbl())
   }
 
-  map_dfr(object$steps, varying_args, full = full)
+  map(object$steps, varying_args, full = full) %>% purrr::list_rbind()
 }
 
 #' @export