From be40f648fc6a9aef99c75d595ae310062653a2cc Mon Sep 17 00:00:00 2001 From: Lukas Burk Date: Wed, 11 Oct 2023 15:14:00 +0200 Subject: [PATCH] fix: ranger did not receive num.threads = 1 during () (#294) --- 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) } )