Update R/recipe.R

Co-authored-by: Emil Hvitfeldt <[email protected]>
rstudio · Jul 19, 2024 · accc0c7 · accc0c7
1 parent e38e9de
commit accc0c7
Showing 1 changed file with 1 addition and 2 deletions.
diff --git a/R/recipe.R b/R/recipe.R
@@ -16,8 +16,7 @@ vetiver_create_meta.recipe <- function(model, metadata) {
 #' @rdname vetiver_create_ptype
 #' @export
 vetiver_ptype.recipe <- function(model, ...) {
-    preds <- vec_slice(model$var_info, model$var_info$role == "predictor")
-    model$ptype[preds$variable]
+    recipes::recipes_ptype(model, stage = "bake")
 }
 
 #' @rdname vetiver_create_description