Support class model_fit directly #1194

vincentarelbundock · Aug 22, 2024 · 23fcddc · 23fcddc
1 parent 77a0c21
commit 23fcddc
Showing 1 changed file with 6 additions and 0 deletions.
diff --git a/R/get_modeldata.R b/R/get_modeldata.R
@@ -9,6 +9,12 @@ get_modeldata <- function(model, additional_variables = FALSE, modeldata = NULL,
     # some pre-processing, and we want to rely on the workflow to do that.
     # workflows are triggered on `stats::predict()`
     if (inherits(model, c("model_fit", "workflow"))) {
+        if ("fit" %in% names(model)) {
+            tmp <- try(get_modeldata(model$fit), silent = TRUE)
+            if (inherits(tmp, "data.frame")) {
+                return(tmp)
+            }
+        }
         return(NULL)
     }