include postprocessor in tune_args.workflow() output (#270)

tidymodels · Nov 15, 2024 · cd34921 · cd34921
1 parent f9d081a
commit cd34921
Showing 1 changed file with 6 additions and 0 deletions.
diff --git a/R/generics.R b/R/generics.R
@@ -40,6 +40,12 @@ tune_args.workflow <- function(object, ...) {
     param_data <- vctrs::vec_rbind(param_data, recipe_param_data)
   }
 
+  if (has_postprocessor_tailor(object)) {
+    tailor <- extract_postprocessor(object)
+    tailor_param_data <- generics::tune_args(tailor)
+    param_data <- vctrs::vec_rbind(param_data, tailor_param_data)
+  }
+
   param_data
 }