diff --git a/torchdata/nodes/loader.py b/torchdata/nodes/loader.py index 7543cb4f8..fad0813c3 100644 --- a/torchdata/nodes/loader.py +++ b/torchdata/nodes/loader.py @@ -123,7 +123,7 @@ def next(self): self._cached_state_dict = None else: item = next(self.root) - self._num_yielded += 1 + self._num_yielded += 1 return item def get_state(self) -> Dict[str, Any]: