Skip to content

Commit bd29426

Browse files
authored
Fix failing tests on old pytorch versions
1 parent a5ee7ae commit bd29426

File tree

1 file changed

+1
-1
lines changed

1 file changed

+1
-1
lines changed

tests/ignite/engine/test_create_supervised.py

Lines changed: 1 addition & 1 deletion
Original file line numberDiff line numberDiff line change
@@ -481,7 +481,7 @@ def test_create_supervised_trainer_on_cuda():
481481
_test_create_mocked_supervised_trainer(model_device=model_device, trainer_device=trainer_device)
482482

483483

484-
@pytest.mark.skipif(not torch.backends.mps.is_available(), reason="Skip if no MPS")
484+
@pytest.mark.skipif(not (_torch_version_le_112 and torch.backends.mps.is_available()), reason="Skip if no MPS")
485485
def test_create_supervised_trainer_on_mps():
486486
model_device = trainer_device = "mps"
487487
_test_create_supervised_trainer_wrong_accumulation(model_device=model_device, trainer_device=trainer_device)

0 commit comments

Comments
 (0)