diff --git a/tests/unittests/bases/test_ddp.py b/tests/unittests/bases/test_ddp.py index 8f1920b8bf2..ad29b0f4c6e 100644 --- a/tests/unittests/bases/test_ddp.py +++ b/tests/unittests/bases/test_ddp.py @@ -339,6 +339,7 @@ def _test_sync_with_unequal_size_lists(rank): @pytest.mark.DDP @pytest.mark.skipif(not _TORCH_GREATER_EQUAL_2_1, reason="test only works on newer torch versions") @pytest.mark.skipif(sys.platform == "win32", reason="DDP not available on windows") +@pytest.mark.skipif(not USE_PYTEST_POOL, reason="DDP pool is not available.") def test_sync_with_unequal_size_lists(): """Test that synchronization of states can be enabled and disabled for compute.""" pytest.pool.map(_test_sync_with_unequal_size_lists, range(NUM_PROCESSES))