fix: fallback predict type warning

mlr-org · Jan 28, 2025 · 5876ff2 · 5876ff2
1 parent bc8b061
commit 5876ff2
Showing 1 changed file with 3 additions and 1 deletion.
diff --git a/R/train_auto.R b/R/train_auto.R
@@ -75,7 +75,9 @@ train_auto = function(self, private, task) {
   graph_learner = as_learner(self$graph)
   graph_learner$id = "graph_learner"
   graph_learner$predict_type = pv$measure$predict_type
-  graph_learner$encapsulate("callr", lrn(paste0(graph_learner$task_type, ".featureless")))
+  fallback = lrn(paste0(graph_learner$task_type, ".featureless"))
+  fallback$predict_type = pv$measure$predict_type
+  graph_learner$encapsulate(method = "callr", fallback = fallback)
   graph_learner$timeout = c(train = pv$learner_timeout, predict = pv$learner_timeout)
 
   learners_with_validation = intersect(learner_ids, c("xgboost", "catboost", "lightgbm"))