Set XLA example time to 500s

2485b8b0 · Lysandre · 2dba7d57 · 2485b8b0
Commit 2485b8b0 authored Oct 15, 2020 by Lysandre
Show whitespace changes
Inline Side-by-side

Showing with 1 addition and 1 deletion

examples/test_xla_examples.py examples/test_xla_examples.py +1 -1

No files found.
--- a/examples/test_xla_examples.py
+++ b/examples/test_xla_examples.py
@@ -80,7 +80,7 @@ class TorchXLAExamplesTests(unittest.TestCase):
                self.assertGreaterEqual(value, 0.70)

            # Assert that the script takes less than 300 seconds to make sure it doesn't hang.
-            self.assertLess(end - start, 300)
+            self.assertLess(end - start, 500)

    def test_trainer_tpu(self):
        import xla_spawn