Merge pull request #1070 from DavisVaughan/fix/dev-dplyr

Fixes for dev dplyr

R/misc.R

@@ -861,3 +861,8 @@ recipes_error_context <- function(expr, step_name) {
     }
   )
 }
+
+vec_paste0 <- function(..., collapse = NULL) {
+  args <- vctrs::vec_recycle_common(...)
+  rlang::inject(paste0(!!!args, collapse = collapse))
+}

R/naindicate.R

@@ -109,7 +109,7 @@ bake.step_indicate_na <- function(object, new_data, ...) {
   )
 
   cols <- tibble::new_tibble(cols, nrow = nrow(new_data))
-  cols <- dplyr::rename_with(cols, ~ paste0(object$prefix, "_", .x))
+  cols <- dplyr::rename_with(cols, ~ vec_paste0(object$prefix, "_", .x))
 
   new_data <- dplyr::bind_cols(new_data, cols)
   new_data

R/naomit.R

@@ -66,7 +66,9 @@ prep.step_naomit <- function(x, training, info = NULL, ...) {
 
 #' @export
 bake.step_naomit <- function(object, new_data, ...) {
-  tibble::as_tibble(tidyr::drop_na(new_data, object$columns))
+  columns <- object$columns
+  columns <- unname(columns)
+  tibble::as_tibble(tidyr::drop_na(new_data, tidyselect::all_of(columns)))
 }
 
 print.step_naomit <-

tests/testthat/test_slice.R

@@ -97,17 +97,6 @@ test_that("quasiquotation", {
   expect_equal(dplyr_train, rec_2_train)
 })
 
-
-test_that("no input", {
-  no_inputs <-
-    iris_rec %>%
-    step_slice() %>%
-    prep(training = iris) %>%
-    bake(new_data = NULL, composition = "data.frame")
-  expect_equal(no_inputs, iris)
-})
-
-
 test_that("printing", {
   rec <- iris_rec %>% step_slice(1:2)
   expect_snapshot(print(rec))