From 5876ff2571bc1a86a4f64aca6a936782f2073d5a Mon Sep 17 00:00:00 2001 From: be-marc Date: Tue, 28 Jan 2025 17:00:26 +0100 Subject: [PATCH] fix: fallback predict type warning --- R/train_auto.R | 4 +++- 1 file changed, 3 insertions(+), 1 deletion(-) diff --git a/R/train_auto.R b/R/train_auto.R index b9fb969..74aff96 100644 --- 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"))