From fc0c2ae8765e5f0785fd2ed66dc2b4652c05b727 Mon Sep 17 00:00:00 2001 From: vfdev Date: Fri, 24 Nov 2023 11:28:51 +0100 Subject: [PATCH] Fix failing tests on old pytorch versions (#3147) --- 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 48a703d60a12..c8230cb58b13 100644 --- 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)