diff --git a/tests/trainer/test_trainer.py b/tests/trainer/test_trainer.py index 31b2d86243..5d8b251736 100755 --- a/tests/trainer/test_trainer.py +++ b/tests/trainer/test_trainer.py @@ -26,6 +26,7 @@ world_size) +@pytest.mark.timeout(30) # TODO lower the timeout. See https://github.com/mosaicml/composer/issues/774. class TestTrainerInit(): @pytest.fixture @@ -114,7 +115,7 @@ def test_init_with_max_duration_in_batches(self, config): @world_size(1, 2) @device('cpu', 'gpu', 'gpu-amp', precision=True) -@pytest.mark.timeout(5) +@pytest.mark.timeout(30) # TODO lower the timeout. See https://github.com/mosaicml/composer/issues/774. class TestTrainerEquivalence(): reference_model: Model @@ -261,6 +262,7 @@ def after_forward(self, state, logger): assert not torch.allclose(original_outputs[0], state.outputs[0]) +@pytest.mark.timeout(30) # TODO lower the timeout. See https://github.com/mosaicml/composer/issues/774. class TestTrainerEvents(): @pytest.fixture @@ -309,6 +311,7 @@ def test_data_not_augmented(self, config): """ +@pytest.mark.timeout(30) # TODO lower the timeout. See https://github.com/mosaicml/composer/issues/774. class TestTrainerAssets: @pytest.fixture