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", {