From 8687901af8e65952ab960daa4d1237dcbfd1a667 Mon Sep 17 00:00:00 2001 From: mb706 Date: Fri, 23 Aug 2024 14:57:26 +0200 Subject: [PATCH] adapt tests --- tests/testthat/test_GraphLearner.R | 2 ++ 1 file changed, 2 insertions(+) diff --git a/tests/testthat/test_GraphLearner.R b/tests/testthat/test_GraphLearner.R index f992aefbd..e4113f505 100644 --- 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