adapt tests

mlr-org · Aug 23, 2024 · 8687901 · 8687901
1 parent 511f846
commit 8687901
Showing 1 changed file with 2 additions and 0 deletions.
diff --git a/tests/testthat/test_GraphLearner.R b/tests/testthat/test_GraphLearner.R
@@ -394,6 +394,8 @@ test_that("graphlearner predict type inference", {
   lrn = GraphLearner$new(po("branch", 2) %>>% gunion(list(lrr, lfr)) %>>% po("unbranch"))
   expect_equal(lrn$predict_type, "response")
   lrn = GraphLearner$new(po("branch", 2) %>>% gunion(list(lrp, lfr)) %>>% po("unbranch"))
+  expect_equal(lrn$predict_type, "prob")
+  lrn$param_set$values$branch.selection = 2
   expect_equal(lrn$predict_type, "response")
 
   # with additional NOP in branch