fix select all warning

jpquast · Nov 4, 2024 · 3df6bb5 · 3df6bb5
1 parent 989fe45
commit 3df6bb5
Showing 1 changed file with 2 additions and 2 deletions.
diff --git a/R/impute_randomforest.R b/R/impute_randomforest.R
@@ -135,8 +135,8 @@ impute_randomforest <- function(
     result <- data_imputed %>%
       dplyr::left_join(
         data %>%
-          dplyr::select(retain_columns, !!sample_sym),
-        by = as.character(sample_sym)
+          dplyr::select(all_of(retain_columns), !!sample_sym),
+        by = c(as.character(sample_sym), as.character(grouping_sym))
       )
   } else {
     result <- data_imputed