ignore these commits, i'm just syncing the origin so i can test on my…

… cloud workstation
byi8220 · Sep 3, 2024 · 7092981 · 7092981
1 parent dff2666
commit 7092981
Showing 1 changed file with 1 addition and 2 deletions.
diff --git a/src/accelerate/test_utils/scripts/test_distributed_data_loop.py b/src/accelerate/test_utils/scripts/test_distributed_data_loop.py
@@ -412,8 +412,7 @@ def main():
 
     # Dataloader after pickling
     iterable_loader = create_dataloader(accelerator, dataset_size=NUM_ELEMENTS, batch_size=BATCH_SIZE, iterable=True)
-    loader = DataLoader(DummyIterableDataset(iterable_loader), shuffle=False, batch_size=BATCH_SIZE, num_workers=NUM_WORKERS)
-    test_pickled_dataloader(loader, accelerator)
+    test_pickled_dataloader(iterable_loader, accelerator)
 
     accelerator.end_training()