diff --git a/tests/ignite/handlers/test_param_scheduler.py b/tests/ignite/handlers/test_param_scheduler.py index 29ecd3a590c..6a323003078 100644 --- a/tests/ignite/handlers/test_param_scheduler.py +++ b/tests/ignite/handlers/test_param_scheduler.py @@ -60,9 +60,7 @@ def test_cyclical_scheduler_asserts(): tensor = torch.zeros([1], requires_grad=True) optimizer = torch.optim.SGD([tensor], lr=0) - with pytest.raises( - TypeError, match="Can't instantiate abstract class CyclicalScheduler with abstract methods _get_cycle_param" - ): + with pytest.raises(TypeError, match="Can't instantiate abstract class CyclicalScheduler"): CyclicalScheduler({}, "lr", 0.0, 1.0, 10)