diff --git a/tests/unittests/wrappers/test_tracker.py b/tests/unittests/wrappers/test_tracker.py index a2981d2d126..9485cf0e276 100644 --- a/tests/unittests/wrappers/test_tracker.py +++ b/tests/unittests/wrappers/test_tracker.py @@ -41,6 +41,11 @@ def test_raises_error_on_wrong_input(): ): MetricTracker(MetricCollection([MeanAbsoluteError(), MeanSquaredError()]), maximize=[False, False, False]) + with pytest.raises( + ValueError, match="Argument `maximize` should be a single bool when `metric` is a single Metric" + ): + MetricTracker(MeanAbsoluteError(), maximize=[False]) + @pytest.mark.parametrize( "method, method_input",