From fc7343aeb84ed8f20959dde81fc4dd62d0488971 Mon Sep 17 00:00:00 2001 From: Mohammad Adil Date: Fri, 6 Aug 2021 01:30:18 -0700 Subject: [PATCH] Increase timeout of tests Signed-off-by: Mohammad Adil --- tests/test_highresnet.py | 2 +- tests/test_integration_determinism.py | 2 +- 2 files changed, 2 insertions(+), 2 deletions(-) diff --git a/tests/test_highresnet.py b/tests/test_highresnet.py index 83248ad85f..61af529b63 100644 --- a/tests/test_highresnet.py +++ b/tests/test_highresnet.py @@ -53,7 +53,7 @@ def test_shape(self, input_param, input_shape, expected_shape): result = net.forward(torch.randn(input_shape).to(device)) self.assertEqual(result.shape, expected_shape) - @TimedCall(seconds=400, force_quit=True) + @TimedCall(seconds=800, force_quit=True) def test_script(self): input_param, input_shape, expected_shape = TEST_CASE_1 net = HighResNet(**input_param) diff --git a/tests/test_integration_determinism.py b/tests/test_integration_determinism.py index 4947610484..e077420420 100644 --- a/tests/test_integration_determinism.py +++ b/tests/test_integration_determinism.py @@ -75,7 +75,7 @@ def setUp(self): def tearDown(self): set_determinism(seed=None) - @TimedCall(seconds=30) + @TimedCall(seconds=150) def test_training(self): set_determinism(seed=0) loss, step = run_test(device=self.device)