From 9d4bb97ca5e6f19ec80ac74bc1a76032d443ea07 Mon Sep 17 00:00:00 2001 From: Emil Hvitfeldt Date: Tue, 18 Jul 2023 14:38:32 -0700 Subject: [PATCH] pass correct col_names --- R/geodist.R | 2 +- R/indicate_na.R | 2 +- 2 files changed, 2 insertions(+), 2 deletions(-) diff --git a/R/geodist.R b/R/geodist.R index fce52f531..b9a14507b 100644 --- a/R/geodist.R +++ b/R/geodist.R @@ -248,7 +248,7 @@ bake.step_geodist <- function(object, new_data, ...) { geo_data <- check_name(geo_data, new_data, object, newname = object$name) new_data <- vec_cbind(new_data, geo_data) - new_data <- remove_original_cols(new_data, object, names(object$columns)) + new_data <- remove_original_cols(new_data, object, col_names) new_data } diff --git a/R/indicate_na.R b/R/indicate_na.R index e9688df55..4c0289735 100644 --- a/R/indicate_na.R +++ b/R/indicate_na.R @@ -117,7 +117,7 @@ bake.step_indicate_na <- function(object, new_data, ...) { cols <- check_name(cols, new_data, object, names(cols)) new_data <- vec_cbind(new_data, cols) - new_data <- remove_original_cols(new_data, object, names(object$columns)) + new_data <- remove_original_cols(new_data, object, col_names) new_data }