Update R/new_epipredict_steps/layer_yeo_johnson.R

dshemetov · dsweber2 · web-flow · commit f644a52da8fb · 2025-03-20T18:08:32.000-07:00
Co-authored-by: David Weber &lt;david.weber2@pm.me&gt;
diff --git a/R/new_epipredict_steps/layer_yeo_johnson.R b/R/new_epipredict_steps/layer_yeo_johnson.R
@@ -155,7 +155,8 @@ slather.layer_epi_YeoJohnson <- function(object, components, workflow, new_data,
     # c(".pred_ahead_1_case_rate", ".pred_ahead_7_case_rate") to
     # c("lambda_ahead_1_case_rate", "lambda_ahead_7_case_rate").
     original_outcome_cols <- str_match(col_names, ".pred_ahead_\\d+_(.*)")[, 2]
-    if (all(original_outcome_cols %nin% names(components$mold$outcomes))) {
+    outcomes_wout_ahead <- str_match(names(components$mold$outcomes), "ahead_\\d+_(.*)")[,2]
+    if (all(original_outcome_cols %nin% outcomes_wout_ahead)) {
       cli_abort("All columns specified in `...` must be outcome columns.", call = rlang::caller_env())
     }
 

Original file line number	Diff line number	Diff line change
`@@ -155,7 +155,8 @@ slather.layer_epi_YeoJohnson <- function(object, components, workflow, new_data,`
`155`	`155`	`# c(".pred_ahead_1_case_rate", ".pred_ahead_7_case_rate") to`
`156`	`156`	`# c("lambda_ahead_1_case_rate", "lambda_ahead_7_case_rate").`
`157`	`157`	`original_outcome_cols <- str_match(col_names, ".pred_ahead_\\d+_(.*)")[, 2]`
`158`		`- if (all(original_outcome_cols %nin% names(components$mold$outcomes))) {`
	`158`	`+ outcomes_wout_ahead <- str_match(names(components$mold$outcomes), "ahead_\\d+_(.*)")[,2]`
	`159`	`+ if (all(original_outcome_cols %nin% outcomes_wout_ahead)) {`
`159`	`160`	cli_abort("All columns specified in `...` must be outcome columns.", call = rlang::caller_env())
`160`	`161`	`}`
`161`	`162`