make BART training compatible for windows systems

mlr-org · Nov 6, 2024 · 1417d6a · 1417d6a
1 parent e09b946
commit 1417d6a
Showing 1 changed file with 3 additions and 1 deletion.
diff --git a/R/learner_BART_surv_bart.R b/R/learner_BART_surv_bart.R
@@ -137,9 +137,11 @@ LearnerSurvLearnerSurvBART = R6Class("LearnerSurvLearnerSurvBART",
       times = truth[, 1]
       delta = truth[, 2] # delta => status
 
+      .fun = ifelse(.Platform$OS.type == "windows", BART::surv.bart, BART::mc.surv.bart)
+
       list(
         model = invoke(
-          BART::mc.surv.bart,
+          .fun,
           x.train = x.train,
           times = times,
           delta = delta,