only separate categories in predict when type is not response

ropensci · Nov 22, 2023 · c2d0d88 · c2d0d88
1 parent f3b32f6
commit c2d0d88
Showing 1 changed file with 2 additions and 1 deletion.
diff --git a/R/predict_helpers.R b/R/predict_helpers.R
@@ -223,7 +223,8 @@ parse_funs <- function(object, type, funs, categories) {
       name <- paste0(fun_names[j], "_", funs_names[i])
       if (length(resp_idx) > 0L) {
         category <- ifelse_(
-          is_categorical(family_stoch[[resp_idx]]),
+          !identical(type, "response") &&
+            is_categorical(family_stoch[[resp_idx]]),
           paste0("_", categories[[funs_names[i]]]),
           ""
         )