From 82f3d78ef88a9fe0cfa9709b4ea9d95690756388 Mon Sep 17 00:00:00 2001 From: be-marc Date: Fri, 6 Sep 2024 18:57:41 +0200 Subject: [PATCH 1/2] compatibility: mlr3 0.21.0 --- NAMESPACE | 1 + R/mlr_callbacks.R | 10 +++++++++- R/zzz.R | 2 +- man/AutoFSelector.Rd | 2 +- 4 files changed, 12 insertions(+), 3 deletions(-) diff --git a/NAMESPACE b/NAMESPACE index 2dff5caf..81cfe545 100644 --- a/NAMESPACE +++ b/NAMESPACE @@ -66,3 +66,4 @@ importFrom(stats,sd) importFrom(utils,bibentry) importFrom(utils,combn) importFrom(utils,head) +importFrom(utils,packageVersion) diff --git a/R/mlr_callbacks.R b/R/mlr_callbacks.R index ea2d26cf..dab12139 100644 --- a/R/mlr_callbacks.R +++ b/R/mlr_callbacks.R @@ -101,7 +101,15 @@ load_callback_svm_rfe = function() { learner_rfe$param_set$values = params learner_rfe$id = learner$id learner_rfe$predict_type = learner$predict_type - learner_rfe$fallback = learner$fallback + + fallback = learner$fallback + if (packageVersion("mlr3") > "0.20.2") { + method = unname(learner$encapsulation[1]) + learner_rfe$encapsulate(method, fallback) + } else { + learner_rfe$encapsulation = learner$encapsulation + learner_rfe$fallback = fallback + } learner_rfe$timeout = learner$timeout learner_rfe$parallel_predict = learner$parallel_predict context$instance$objective$learner = learner_rfe diff --git a/R/zzz.R b/R/zzz.R index 35cf5485..501d5f63 100644 --- a/R/zzz.R +++ b/R/zzz.R @@ -5,7 +5,7 @@ #' @import mlr3 #' @import bbotk #' @importFrom R6 R6Class -#' @importFrom utils combn head +#' @importFrom utils combn head packageVersion #' @importFrom stats sd "_PACKAGE" diff --git a/man/AutoFSelector.Rd b/man/AutoFSelector.Rd index a7d46be4..dd4954de 100644 --- a/man/AutoFSelector.Rd +++ b/man/AutoFSelector.Rd @@ -149,7 +149,7 @@ Hash (unique identifier) for this partial object, excluding some components whic \if{html}{\out{
Inherited methods