Increase the timeout for test trainer equivalence (#766)

It occasionally fails on Jenkins with a 2 second timeout. Increasing to 5 seconds.
mosaicml · Mar 16, 2022 · 1c1d8c9 · 1c1d8c9
1 parent 80c537c
commit 1c1d8c9
Showing 1 changed file with 1 addition and 0 deletions.
diff --git a/tests/trainer/test_trainer.py b/tests/trainer/test_trainer.py
@@ -114,6 +114,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)
 class TestTrainerEquivalence():
 
     reference_model: Model