Fix failing tests on old pytorch versions (#3147)

pytorch · Nov 24, 2023 · fc0c2ae · fc0c2ae
1 parent e5997ee
commit fc0c2ae
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)