run precommit

pytorch · Feb 11, 2025 · 10ba260 · 10ba260
1 parent 6b7a106
commit 10ba260
Showing 1 changed file with 1 addition and 1 deletion.
diff --git a/torchdata/stateful_dataloader/sampler.py b/torchdata/stateful_dataloader/sampler.py
@@ -60,7 +60,7 @@ def __next__(self):
                 if self.perm is None or not self.perm:
                     self.perm = torch.randint(
                         high=self.n,
-                        size=(self.remainder,),
+                        size=(remainder,),
                         dtype=torch.int64,
                         generator=self.generator,
                     ).tolist()