Merge pull request #203 from tidymodels/strings_as_factors

tidymodels · Oct 13, 2023 · e679e81 · e679e81
2 parents cd8d730 + 079c887
commit e679e81
Show file tree

Hide file tree

Showing 2 changed files with 5 additions and 1 deletion.
diff --git a/NEWS.md b/NEWS.md
@@ -1,5 +1,7 @@
 # embed (development version)
 
+* `step_collapse_stringdist()` will now return predictors as factors.
+
 # embed 1.1.2
 
 ## Improvements

diff --git a/R/collapse_stringdist.R b/R/collapse_stringdist.R
@@ -194,7 +194,9 @@ bake.step_collapse_stringdist <- function(object, new_data, ...) {
 collapse_apply <- function(x, dict) {
   dict <- purrr::map_dfr(dict, ~ list(from = .x, to = .x[1]))
 
-  dict$to[match(x, dict$from)]
+  res <- dict$to[match(x, dict$from)]
+
+  factor(res, levels = unique(dict$to))
 }
 
 #' @export