From d20aea4723486b4a6f331f17d43b6526f17ad8fc Mon Sep 17 00:00:00 2001
From: SeanNaren <sean@grid.ai>
Date: Tue, 13 Apr 2021 15:16:57 +0100
Subject: [PATCH] Revert limit train batches

---
 tests/models/test_tpu.py | 2 +-
 1 file changed, 1 insertion(+), 1 deletion(-)

diff --git a/tests/models/test_tpu.py b/tests/models/test_tpu.py
index 566c222e3d01b6..261adf5bc45af4 100644
--- a/tests/models/test_tpu.py
+++ b/tests/models/test_tpu.py
@@ -122,7 +122,7 @@ def test_model_16bit_tpu_cores_1(tmpdir):
         progress_bar_refresh_rate=0,
         max_epochs=2,
         tpu_cores=1,
-        limit_train_batches=8,
+        limit_train_batches=0.7,
         limit_val_batches=2,
     )