Skip to content

Commit 1417d6a

Browse files
committed
make BART training compatible for windows systems
1 parent e09b946 commit 1417d6a

File tree

1 file changed

+3
-1
lines changed

1 file changed

+3
-1
lines changed

R/learner_BART_surv_bart.R

Lines changed: 3 additions & 1 deletion
Original file line numberDiff line numberDiff line change
@@ -137,9 +137,11 @@ LearnerSurvLearnerSurvBART = R6Class("LearnerSurvLearnerSurvBART",
137137
times = truth[, 1]
138138
delta = truth[, 2] # delta => status
139139

140+
.fun = ifelse(.Platform$OS.type == "windows", BART::surv.bart, BART::mc.surv.bart)
141+
140142
list(
141143
model = invoke(
142-
BART::mc.surv.bart,
144+
.fun,
143145
x.train = x.train,
144146
times = times,
145147
delta = delta,

0 commit comments

Comments
 (0)