refactor: read task features from state (#278)

* refactor: read task features from state * refactor: feature_names
mlr-org · Dec 21, 2023 · e052003 · e052003
1 parent 86f19eb
commit e052003
Showing 1 changed file with 1 addition and 1 deletion.
diff --git 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$data_prototype)
+  cols = names(learner$state$data_prototype) %??% learner$state$feature_names
   task$data(cols = intersect(cols, task$feature_names))
 }