remove unnecessary comment

pytorch · Feb 5, 2025 · 1288e77 · 1288e77
1 parent 6ba9d94
commit 1288e77
Showing 1 changed file with 0 additions and 1 deletion.
diff --git a/torchdata/stateful_dataloader/sampler.py b/torchdata/stateful_dataloader/sampler.py
@@ -85,7 +85,6 @@ def __next__(self) -> list[int]:
             return batch
         except StopIteration:
             if self.drop_last or len(batch) == 0:
-                # Reset the iterator for the next epoch
                 raise StopIteration
             else:
                 return batch