From 038fec14d83f7d09b8742c4d517368f12a1392ae Mon Sep 17 00:00:00 2001 From: Sebastian Fischer Date: Fri, 26 Jan 2024 10:52:31 +0100 Subject: [PATCH] fix tests for featureless --- tests/testthat/test_LearnerTorchFeatureless.R | 4 ++-- 1 file changed, 2 insertions(+), 2 deletions(-) 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")) })