fix checks

abichat · Mar 6, 2024 · fa30fc7 · fa30fc7
1 parent e5c63c2
commit fa30fc7
Show file tree

Hide file tree

Showing 3 changed files with 16 additions and 2 deletions.
diff --git a/NAMESPACE b/NAMESPACE
@@ -39,6 +39,7 @@ export(step_select_kruskal)
 export(step_select_wilcoxon)
 export(var_to_keep)
 importFrom(dplyr,filter)
+importFrom(dplyr,if_else)
 importFrom(dplyr,left_join)
 importFrom(dplyr,mutate)
 importFrom(dplyr,pull)
@@ -56,6 +57,7 @@ importFrom(recipes,recipes_eval_select)
 importFrom(recipes,sel2char)
 importFrom(recipes,step)
 importFrom(rlang,.data)
+importFrom(rlang,.env)
 importFrom(rlang,abort)
 importFrom(rlang,enquos)
 importFrom(stats,as.formula)

diff --git a/R/aggregate_list.R b/R/aggregate_list.R
@@ -107,8 +107,9 @@ step_aggregate_list_new <- function(terms, role, trained,
 }
 
 #' @export
-#' @importFrom dplyr left_join
+#' @importFrom dplyr if_else left_join
 #' @importFrom recipes recipes_eval_select
+#' @importFrom rlang .data .env
 #' @importFrom tibble enframe tibble
 #' @importFrom tidyr unnest_longer
 prep.step_aggregate_list <- function(x, training, info = NULL, ...) {
@@ -131,7 +132,8 @@ prep.step_aggregate_list <- function(x, training, info = NULL, ...) {
     asis <- setdiff(col_names, unlist(x$list_agg))
     res_agg_list <-
       res_agg_list %>%
-      mutate(aggregate = if_else(terms %in% asis, terms, aggregate))
+      mutate(aggregate = if_else(.data$terms %in% .env$asis,
+                                 .data$terms, .data$aggregate))
   }
 
   step_aggregate_list_new(

diff --git a/man/step_aggregate_list.Rd b/man/step_aggregate_list.Rd