From accc0c79a6bf448c68728aa0fb6320d39dccba29 Mon Sep 17 00:00:00 2001
From: Julia Silge <julia.silge@gmail.com>
Date: Fri, 19 Jul 2024 13:00:30 -0600
Subject: [PATCH] Update R/recipe.R

Co-authored-by: Emil Hvitfeldt <emilhhvitfeldt@gmail.com>
---
 R/recipe.R | 3 +--
 1 file changed, 1 insertion(+), 2 deletions(-)

diff --git a/R/recipe.R b/R/recipe.R
index 9ab56942..f6e73a2a 100644
--- 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