Km issue #22: unite

NAMESPACE

@@ -1,16 +1,13 @@
 # Generated by roxygen2: do not edit by hand
 
-S3method(bake,step_epi_ahead)
 S3method(bake,step_epi_lag)
 S3method(epi_keys,default)
 S3method(epi_keys,epi_df)
 S3method(epi_keys,recipe)
 S3method(epi_recipe,default)
 S3method(epi_recipe,epi_df)
 S3method(epi_recipe,formula)
-S3method(prep,step_epi_ahead)
 S3method(prep,step_epi_lag)
-S3method(print,step_epi_ahead)
 S3method(print,step_epi_lag)
 export("%>%")
 export(arx_args_list)
@@ -27,7 +24,6 @@ export(knnarx_args_list)
 export(knnarx_forecaster)
 export(smooth_arx_args_list)
 export(smooth_arx_forecaster)
-export(step_epi_ahead)
 export(step_epi_lag)
 import(recipes)
 importFrom(magrittr,"%>%")

R/epi_lag.R

@@ -16,13 +16,13 @@
 #'
 #' @family row operation steps
 #' @export
-#' @rdname step_epi_ahead
+#' @rdname step_epi_lag
 step_epi_lag <-
   function(recipe,
            ...,
            role = "predictor",
            trained = FALSE,
-           lag = 1,
+           lag = 1, # negative for ahead
            prefix = "lag_",
            default = NA,
            keys = epi_keys(recipe),
@@ -117,7 +117,7 @@ bake.step_epi_lag <- function(object, new_data, ...) {
 print.step_epi_lag <-
   function(x, width = max(20, options()$width - 30), ...) {
     ## TODO add printing of the lags
-    title <- "Lagging "
+    title <- ifelse(x$lag >= 0, "Lagging", "Leading")
     recipes::print_step(x$columns, x$terms, x$trained, title, width)
     invisible(x)
   }

musings/example-recipe.R

@@ -36,7 +36,7 @@ xx <- x %>% filter(time_value > "2021-12-01")
 # Baseline AR3
 r <- epi_recipe(x) %>% # if we add this as a class, maybe we get better
                        # behaviour downstream?
-  step_epi_ahead(death_rate, ahead = 7) %>%
+  step_epi_lag(death_rate, lag = -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()) %>%

tests/testthat/test-assign_arg_list.R

@@ -0,0 +1,6 @@
+test_that("First argument must be a list",{
+  expect_error(assign_arg_list(c(1,2,3)))
+})
+test_that("All arguments should be named",{
+  expect_error(assign_arg_list(list(1,2)))
+})