Merge pull request #335 from mlr-org/fix_c_distrs

Fix combining distrs
mlr-org · Oct 19, 2023 · e557cd7 · e557cd7
2 parents 5a92942 + cf5a369
commit e557cd7
Show file tree

Hide file tree

Showing 2 changed files with 3 additions and 2 deletions.
diff --git a/DESCRIPTION b/DESCRIPTION
@@ -43,7 +43,7 @@ Depends:
 Imports:
     checkmate,
     data.table,
-    distr6 (>= 1.8.2),
+    distr6 (>= 1.8.3),
     ggplot2,
     mlr3misc (>= 0.7.0),
     mlr3viz,

diff --git a/R/PredictionDataSurv.R b/R/PredictionDataSurv.R
@@ -90,7 +90,8 @@ c.PredictionDataSurv = function(..., keep_duplicates = TRUE) {
 
     # All distributions? Concatenate!
     if (test_dist) {
-      result$distr = do.call(c, distr_list)
+      result$distr = do.call(c, c(distr_list,
+        list(decorators = c("CoreStatistics", "ExoticStatistics"))))
     } else {
       dims = vapply(distr_list, function(.x) length(dim(.x)), integer(1))
       # If mix of arrays and matrices, convert arrays to median survival matrices