fix: attr stuff

Author: dshemetov

R/step_yeo_johnson.R

@@ -166,8 +166,8 @@ prep.step_epi_YeoJohnson <- function(x, training, info = NULL, ...) {
     limits = x$limits,
     num_unique = x$num_unique,
     na_rm = x$na_rm,
-    forecast_date = attributes(training)$metadata$as_of,
-    metadata = attributes(training)$metadata,
+    forecast_date = attr(training, "metadata")$as_of,
+    metadata = attr(training, "metadata"),
     columns = col_names,
     skip = x$skip,
     id = x$id
@@ -179,13 +179,13 @@ bake.step_epi_YeoJohnson <- function(object, new_data, ...) {
   # If not an epi_df, make it one assuming the template of training data.
   # If it is an epi_df, check that the keys match.
   # Imitating the pattern in step_adjust_latency().
-  if (!inherits(new_data, "epi_df") || is.null(attributes(new_data)$metadata$as_of)) {
+  if (!inherits(new_data, "epi_df") || is.null(attr(new_data, "metadata")$as_of)) {
     new_data <- as_epi_df(
       new_data,
       as_of = object$forecast_date,
       other_keys = object$metadata$other_keys %||% character()
     )
-    new_data %@% metadata <- object$metadata
+    attr(new_data, "metadata") <- object$metadata
   }
   # Check that the keys match.
   keys <- key_colnames(new_data, exclude = "time_value")