From 9555e428d56af5db38a2ca775462fe23b4ee20b5 Mon Sep 17 00:00:00 2001 From: Lukas Burk Date: Tue, 10 Oct 2023 15:40:50 +0200 Subject: [PATCH] fix: ranger did not receive num.threads = 1 during () --- R/learner_ranger_surv_ranger.R | 2 +- 1 file changed, 1 insertion(+), 1 deletion(-) diff --git a/R/learner_ranger_surv_ranger.R b/R/learner_ranger_surv_ranger.R index 7f68e6fcf..b471e7965 100644 --- a/R/learner_ranger_surv_ranger.R +++ b/R/learner_ranger_surv_ranger.R @@ -117,7 +117,7 @@ delayedAssign( .predict = function(task) { pv = self$param_set$get_values(tags = "predict") newdata = ordered_features(task, self) - prediction = predict(object = self$model, data = newdata) + prediction = invoke(predict, object = self$model, data = newdata, .args = pv) mlr3proba::.surv_return(times = prediction$unique.death.times, surv = prediction$survival) } )