remove prints

byi8220 · Sep 4, 2024 · 47557b7 · 47557b7
1 parent ee681ed
commit 47557b7
Show file tree

Hide file tree

Showing 2 changed files with 0 additions and 2 deletions.
diff --git a/src/accelerate/data_loader.py b/src/accelerate/data_loader.py
@@ -1212,7 +1212,6 @@ def __iter__(self):
         self.end()
 
     def __len__(self):
-        print("len called")
         return len(self.base_dataloader) - self.skip_batches
 
     def __reduce__(self):

diff --git a/tests/test_accelerator.py b/tests/test_accelerator.py
@@ -686,7 +686,6 @@ def test_can_pickle_dataloader(self, dispatch_batches):
 
         # Test skip dataloader works as expected as well
         loaded_skip_dl = model_loaded._dataloaders[1]
-        print(model_loaded._dataloaders)
         assert isinstance(loaded_skip_dl, DataLoader)
         if dispatch_batches:
             assert isinstance(loaded_dl, DataLoaderDispatcher)