Update R/learner_BART_surv_bart.R

Co-authored-by: Sebastian Fischer <[email protected]>
mlr-org · Aug 21, 2023 · f18d2d9 · f18d2d9
1 parent 387a41e
commit f18d2d9
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
@@ -176,7 +176,7 @@ delayedAssign(
       #' @param type Can be either `count` or `prob`.
       #' @return Named `numeric()`.
       importance = function(type = 'count') {
-        if (!type %in% c('prob', 'count')) stopf("type can be only 'prob' or 'count'")
+assert_choice(type, c("prob", "count"))
 
         if (is.null(self$model)) {
           stopf("No model stored")