extract parameter set from postprocessor

tidymodels · Oct 16, 2024 · c0f38f0 · c0f38f0
1 parent 78aa5df
commit c0f38f0
Showing 1 changed file with 7 additions and 0 deletions.
diff --git a/R/extract.R b/R/extract.R
@@ -225,6 +225,13 @@ extract_parameter_set_dials.workflow <- function(x, ...) {
     param_data <- vctrs::vec_rbind(param_data, recipe_param_data)
   }
 
+  if (has_postprocessor_tailor(x)) {
+    tailor <- extract_postprocessor(x)
+    tailor_param_data <- extract_parameter_set_dials(tailor)
+
+    param_data <- vctrs::vec_rbind(param_data, tailor_param_data)
+  }
+
   dials::parameters_constr(
     param_data$name,
     param_data$id,