Put some work into it.

Author: kenmawer

R/step_naomit2.R

@@ -0,0 +1,12 @@
+#' Special NA omit step that does two steps in one
+#'
+#' @param x Recipe to be used.
+#'
+#' @return A recipe with NA's omitted ....
+#' @export recipes
+
+step_naomit2 <- function(x) {
+  x %>%
+    recipes::step_naomit(all_predictors()) %>%
+    recipes::step_naomit(all_outcomes(), skip = TRUE)
+}

musings/example-recipe.R

@@ -40,10 +40,7 @@ r <- epi_recipe(x) %>% # if we add this as a class, maybe we get better
   step_epi_ahead(death_rate, ahead = 7) %>%
   step_epi_lag(death_rate, lag = c(0, 7, 14)) %>%
   step_epi_lag(case_rate, lag = c(0, 7, 14)) %>%
-  step_naomit(all_predictors()) %>%
-  # below, `skip` means we don't do this at predict time
-  # we should probably do something useful here to avoid user error
-  step_naomit(all_outcomes(), skip = TRUE)
+  step_narm()
 
 
 slm <- linear_reg()