upd test

Author: voorhs

tests/pipeline/test_optimization.py

@@ -109,7 +109,7 @@ def test_validate_search_space_multiclass(dataset):
 
     pipeline_optimizer = Pipeline.from_search_space(search_space)
     with pytest.raises(ValueError, match="Module 'adaptive' does not support multiclass datasets."):
-        pipeline_optimizer.validate_modules(dataset)
+        pipeline_optimizer.validate_modules(dataset, mode="raise")
 
 
 def test_validate_search_space_multilabel(dataset):
@@ -124,4 +124,4 @@ def test_validate_search_space_multilabel(dataset):
     ]
     pipeline_optimizer = Pipeline.from_search_space(search_space)
     with pytest.raises(ValueError, match="Module 'argmax' does not support multilabel datasets."):
-        pipeline_optimizer.validate_modules(dataset)
+        pipeline_optimizer.validate_modules(dataset, mode="raise")