Fix failing tests on old pytorch versions

pytorch · Nov 24, 2023 · bd29426 · bd29426
1 parent a5ee7ae
commit bd29426
Showing 1 changed file with 1 addition and 1 deletion.
diff --git a/tests/ignite/engine/test_create_supervised.py b/tests/ignite/engine/test_create_supervised.py
@@ -481,7 +481,7 @@ def test_create_supervised_trainer_on_cuda():
     _test_create_mocked_supervised_trainer(model_device=model_device, trainer_device=trainer_device)
 
 
-@pytest.mark.skipif(not torch.backends.mps.is_available(), reason="Skip if no MPS")
+@pytest.mark.skipif(not (_torch_version_le_112 and torch.backends.mps.is_available()), reason="Skip if no MPS")
 def test_create_supervised_trainer_on_mps():
     model_device = trainer_device = "mps"
     _test_create_supervised_trainer_wrong_accumulation(model_device=model_device, trainer_device=trainer_device)