Skip to content

Commit 5876ff2

Browse files
committed
fix: fallback predict type warning
1 parent bc8b061 commit 5876ff2

File tree

1 file changed

+3
-1
lines changed

1 file changed

+3
-1
lines changed

Diff for: R/train_auto.R

+3-1
Original file line numberDiff line numberDiff line change
@@ -75,7 +75,9 @@ train_auto = function(self, private, task) {
7575
graph_learner = as_learner(self$graph)
7676
graph_learner$id = "graph_learner"
7777
graph_learner$predict_type = pv$measure$predict_type
78-
graph_learner$encapsulate("callr", lrn(paste0(graph_learner$task_type, ".featureless")))
78+
fallback = lrn(paste0(graph_learner$task_type, ".featureless"))
79+
fallback$predict_type = pv$measure$predict_type
80+
graph_learner$encapsulate(method = "callr", fallback = fallback)
7981
graph_learner$timeout = c(train = pv$learner_timeout, predict = pv$learner_timeout)
8082

8183
learners_with_validation = intersect(learner_ids, c("xgboost", "catboost", "lightgbm"))

0 commit comments

Comments
 (0)