tests: try to fix stochastic error

mlr-org · Jun 16, 2023 · a5a58f5 · a5a58f5
1 parent 916c7bd
commit a5a58f5
Showing 1 changed file with 2 additions and 2 deletions.
diff --git a/tests/testthat/test_mlr_learners_classif_featureless.R b/tests/testthat/test_mlr_learners_classif_featureless.R
@@ -10,14 +10,14 @@ test_that("Simple training/predict", {
   learner = lrn("classif.featureless")
   expect_learner(learner, task)
 
-  learner$train(task)
+  learner$train(task, row_ids = c(1:50, 51:70, 101:120))
   learner$predict(task)
   expect_class(learner$model, "classif.featureless_model")
   expect_numeric(learner$model$tab, len = 3L, any.missing = FALSE)
   prediction = learner$predict(task)
   expect_factor(prediction$response, any.missing = FALSE, levels = levels(iris$Species))
   perf = prediction$score(msr("classif.ce"))
-  expect_number(perf, lower = 0.6, upper = 0.8)
+  expect_number(perf, lower = 0.6, upper = 0.7)
 })
 
 test_that("Predict with prob", {