fix: ranger did not receive num.threads = 1 during () (#294)

mlr-org · Oct 11, 2023 · be40f64 · be40f64
1 parent 6e2af9e
commit be40f64
Showing 1 changed file with 1 addition and 1 deletion.
diff --git 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)
       }
     )