Make as_forecast() a generic

epiforecasts · Dec 16, 2023 · 0b6958e · 0b6958e
1 parent dd1a662
commit 0b6958e
Show file tree

Hide file tree

Showing 3 changed files with 14 additions and 2 deletions.
diff --git a/NAMESPACE b/NAMESPACE
@@ -1,5 +1,6 @@
 # Generated by roxygen2: do not edit by hand
 
+S3method(as_forecast,default)
 S3method(print,scoringutils_check)
 S3method(quantile_to_interval,data.frame)
 S3method(quantile_to_interval,numeric)

diff --git a/R/validate.R b/R/validate.R
@@ -19,7 +19,13 @@
 #' @examples
 #' as_forecast(example_binary)
 #' as_forecast(example_quantile)
-as_forecast <- function(data) {
+as_forecast <- function(data, ...) {
+  UseMethod("as_forecast")
+}
+
+#' @rdname as_forecast
+#' @export
+as_forecast.default <- function(data, ...) {
   assert(check_data_columns(data))
 
   # find forecast type

diff --git a/man/as_forecast.Rd b/man/as_forecast.Rd