From 1e90d978ee70c85b7826fe36d66c2b2f9f32d008 Mon Sep 17 00:00:00 2001 From: Gulin7 Date: Wed, 4 Dec 2024 20:32:28 +0200 Subject: [PATCH] Fix regex in test_create_supervised --- tests/ignite/engine/test_create_supervised.py | 2 +- 1 file changed, 1 insertion(+), 1 deletion(-) diff --git a/tests/ignite/engine/test_create_supervised.py b/tests/ignite/engine/test_create_supervised.py index a10de3cb4b2..92ef6142b6b 100644 --- a/tests/ignite/engine/test_create_supervised.py +++ b/tests/ignite/engine/test_create_supervised.py @@ -457,7 +457,7 @@ def test_create_supervised_trainer_amp_error(mock_torch_cuda_amp_module): _test_create_supervised_trainer_wrong_accumulation(trainer_device="cpu", amp_mode="amp") with pytest.raises(ImportError, match="Please install torch>=1.12.0 to use amp_mode='amp'."): _test_create_supervised_trainer(amp_mode="amp") - with pytest.raises(ImportError, match="Please install torch>=1.12.0 to use amp_mode='amp'."): + with pytest.raises(ImportError, match="Please install torch>=1.6.0 to use scaler argument."): _test_create_supervised_trainer(amp_mode="amp", scaler=True)