Keep setting attributes for data.frames as well

epiforecasts · Sep 28, 2024 · 1885d87 · 1885d87
1 parent 286a1dd
commit 1885d87
Showing 1 changed file with 2 additions and 0 deletions.
diff --git a/R/score.R b/R/score.R
@@ -356,6 +356,8 @@ validate_scores <- function(scores) {
   ret <- NextMethod()
   if (is.data.table(ret)) {
     setattr(ret, "metrics", attr(x, "metrics"))
+  } else if (is.data.frame(ret)) {
+    attr(ret, "metrics") <- attr(x, "metrics")
   }
   return(ret)
 }