Merge pull request #737 from imbs-hl/fix_reg_test

fix regularization test
imbs-hl · Aug 13, 2024 · fe89494 · fe89494
2 parents de75150 + 7be1fce
commit fe89494
Showing 1 changed file with 1 addition and 1 deletion.
diff --git a/tests/testthat/test_regularization.R b/tests/testthat/test_regularization.R
@@ -132,7 +132,7 @@ test_that("Fewer variables used with regularization, survival extratrees", {
   rf_noreg <- ranger(Surv(time, status) ~ ., dat_surv, num.trees = 5, min.node.size = 10, mtry = 4, splitrule = "extratrees")
   rf_reg <- ranger(Surv(time, status) ~ ., dat_surv, num.trees = 5, min.node.size = 10, mtry = 4, splitrule = "extratrees", num.threads = 1, 
                    regularization.factor = .0001, regularization.usedepth = TRUE)
-  expect_lt(get_num_splitvars(rf_reg), 
+  expect_lte(get_num_splitvars(rf_reg), 
             get_num_splitvars(rf_noreg))
 })