everything is broken

Author: dajmcdon

R/epi_recipe.R

@@ -95,7 +95,9 @@ add_epi_recipe <- function(
 #' @rdname add_epi_recipe
 #' @export
 remove_epi_recipe <- function(x) {
-  workflows::remove_recipe(x)
+  wf <- workflows::remove_recipe(x)
+  class(wf) <- c("epi_workflow", class(wf))
+  wf
 }
 
 
@@ -222,6 +224,7 @@ prep.epi_recipe <- function(
   if (!strings_as_factors) {
     return(NextMethod("prep"))
   }
+  browser()
   # workaround to avoid converting strings2factors with recipes::prep.recipe()
   # We do the conversion here, then set it to FALSE
   training <- recipes:::check_training_set(training, x, fresh)

R/epi_workflow.R

@@ -96,7 +96,6 @@ fit.epi_workflow <- function(object, data, ..., control = workflows::control_wor
     as_of = attributes(data)$metadata$as_of
   )
   object$original_data <- data
-
   NextMethod()
 }
 

tests/testthat/test-epi_workflow.R

@@ -59,6 +59,7 @@ test_that("model can be added/updated/removed from epi_workflow", {
   expect_equal(class(model_spec2), c("linear_reg", "model_spec"))
 
   wf <- remove_model(wf)
+  expect_equal(class(wf), c("epi_workflow", "workflow"))
   expect_error(extract_spec_parsnip(wf))
   expect_equal(wf$fit$actions$model$spec, NULL)
 })