diff --git a/tests/testthat/test_LearnerTorchFeatureless.R b/tests/testthat/test_LearnerTorchFeatureless.R index bd0f29a6..5228de86 100644 --- 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")) })