better constraction of 'which.curve' parameter

mlr-org · Oct 6, 2023 · 21e5be9 · 21e5be9
1 parent 6e9bcff
commit 21e5be9
Showing 1 changed file with 1 addition and 1 deletion.
diff --git a/R/learner_BART_surv_bart.R b/R/learner_BART_surv_bart.R
@@ -72,7 +72,7 @@ delayedAssign(
           openmp = p_lgl(default = TRUE, tags = "predict"),
           quiet = p_lgl(default = TRUE, tags = "predict"),
           importance = p_fct(default = "count", levels = c("count", "prob"), tags = "train"),
-          which.curve = p_uty(tags = "predict")
+          which.curve = p_dbl(special_vals = list("mean"), tags = "predict")
         )
 
         # custom defaults