From 60da1d64746d2c3b051c9f2adc845ad4a3be608b Mon Sep 17 00:00:00 2001
From: Michel Lang <michellang@gmail.com>
Date: Thu, 20 Jan 2022 21:11:42 +0100
Subject: [PATCH] chore: depend on new mlr3, remove workaround

---
 DESCRIPTION | 2 +-
 R/helpers.R | 2 +-
 2 files changed, 2 insertions(+), 2 deletions(-)

diff --git a/DESCRIPTION b/DESCRIPTION
index de3a26aa..b181cfa1 100644
--- a/DESCRIPTION
+++ b/DESCRIPTION
@@ -22,7 +22,7 @@ URL: https://mlr3learners.mlr-org.com,
     https://github.com/mlr-org/mlr3learners
 BugReports: https://github.com/mlr-org/mlr3learners/issues
 Depends:
-    mlr3 (>= 0.12.0-9000),
+    mlr3 (>= 0.13.1),
     R (>= 3.1.0)
 Imports:
     checkmate,
diff --git a/R/helpers.R b/R/helpers.R
index 33cdba98..03dc356c 100644
--- a/R/helpers.R
+++ b/R/helpers.R
@@ -8,7 +8,7 @@ pvec2mat = function(p, levs) {
 
 
 ordered_features = function(task, learner) {
-  cols = names(learner$state$task_prototype %??% learner$state$data_prototype)
+  cols = names(learner$state$data_prototype)
   task$data(cols = intersect(cols, task$feature_names))
 }