Two changes Header of Recipe --> Epi Recipe, added print statement for the unexported print.step_naomit() from recipes.

Author: rachlobay

NAMESPACE

@@ -72,6 +72,7 @@ S3method(print,step_epi_ahead)
 S3method(print,step_epi_lag)
 S3method(print,step_growth_rate)
 S3method(print,step_lag_difference)
+S3method(print,step_naomit)
 S3method(print,step_population_scaling)
 S3method(print,step_training_window)
 S3method(quantile,dist_quantiles)

R/epi_recipe.R

@@ -415,7 +415,7 @@ kill_levels <- function(x, keys) {
 print.epi_recipe <- function(x, form_width = 30, ...) {
   cli::cli_div(theme = list(.pkg = list("vec-trunc" = Inf, "vec-last" = ", ")))
 
-  cli::cli_h1("Recipe")
+  cli::cli_h1("Epi Recipe")
   cli::cli_h3("Inputs")
 
   tab <- table(x$var_info$role, useNA = "ifany")

R/step_epi_naomit.R

@@ -17,3 +17,11 @@ step_epi_naomit <- function(recipe) {
     recipes::step_naomit(all_predictors(), skip = FALSE) %>%
     recipes::step_naomit(all_outcomes(), skip = TRUE)
 }
+
+#' @export
+print.step_naomit <- # not exported from recipes package
+  function(x, width = max(20, options()$width - 30), ...) {
+    title <- "Removing rows with NA values in "
+    print_step(x$columns, x$terms, x$trained, title, width)
+    invisible(x)
+  }