two objects of the same class

cmu-delphi · Jun 17, 2023 · 206f0ef · 206f0ef
1 parent ea48355
commit 206f0ef
Show file tree

Hide file tree

Showing 2 changed files with 5 additions and 4 deletions.
diff --git a/NAMESPACE b/NAMESPACE
@@ -51,6 +51,7 @@ S3method(print,arx_class)
 S3method(print,arx_fcast)
 S3method(print,canned_epipred)
 S3method(print,epi_workflow)
+S3method(print,flat_fcast)
 S3method(print,flatline)
 S3method(print,frosting)
 S3method(print,layer_add_forecast_date)

diff --git a/R/flatline_forecaster.R b/R/flatline_forecaster.R
@@ -33,7 +33,7 @@ flatline_forecaster <- function(
     args_list = flatline_args_list()) {
 
   validate_forecaster_inputs(epi_data, outcome, "time_value")
-  if (!inherits(args_list, c("flatline", "alist"))) {
+  if (!inherits(args_list, c("flat_fcast", "alist"))) {
     cli_stop("args_list was not created using `flatline_args_list().")
   }
   keys <- epi_keys(epi_data)
@@ -77,7 +77,7 @@ flatline_forecaster <- function(
       training = attr(epi_data, "metadata"),
       forecast_created = Sys.time()
     )),
-    class = c("flatline", "canned_epipred")
+    class = c("flat_fcast", "canned_epipred")
   )
 }
 
@@ -125,12 +125,12 @@ flatline_args_list <- function(
            symmetrize,
            nonneg,
            quantile_by_key),
-    class = c("flatline", "alist")
+    class = c("flat_fcast", "alist")
   )
 }
 
 #' @export
-print.flatline <- function(x, ...) {
+print.flat_fcast <- function(x, ...) {
   name <- "flatline"
   NextMethod(name = name, ...)
 }