fix tests for featureless

mlr-org · Jan 26, 2024 · 038fec1 · 038fec1
1 parent 9b22539
commit 038fec1
Showing 1 changed file with 2 additions and 2 deletions.
diff --git a/tests/testthat/test_LearnerTorchFeatureless.R b/tests/testthat/test_LearnerTorchFeatureless.R
@@ -11,7 +11,7 @@ test_that("dataset_featureless works", {
 test_that("Basic checks: Classification", {
   learner = lrn("classif.torch_featureless")
   expect_learner_torch(learner)
-  expect_set_equal(learner$properties, c("twoclass", "multiclass", "missings", "featureless"))
+  expect_set_equal(learner$properties, c("twoclass", "multiclass", "missings", "featureless", "bundle"))
 })
 
 test_that("LearnerTorchFeatureless works", {
@@ -28,5 +28,5 @@ test_that("LearnerTorchFeatureless works", {
 test_that("Basic checks: Regression", {
   learner = lrn("regr.torch_featureless")
   expect_learner_torch(learner)
-  expect_set_equal(learner$properties, c("missings", "featureless"))
+  expect_set_equal(learner$properties, c("missings", "featureless", "bundle"))
 })