unused

rohitgr7 · rohitgr7 · commit c3ab1f1491a8 · 2021-10-19T15:48:20.000+05:30
diff --git a/tests/models/test_tpu.py b/tests/models/test_tpu.py
@@ -39,13 +39,6 @@
 
     SERIAL_EXEC = xmp.MpSerialExecutor()
 
-_LARGER_DATASET = RandomDataset(32, 2000)
-
-
-# 8 cores needs a big dataset
-def _serial_train_loader():
-    return DataLoader(_LARGER_DATASET, batch_size=32)
-
 
 class SerialLoaderBoringModel(BoringModel):
     def train_dataloader(self):
