fix_loader_state (pytorch#1457)

divyanshk · web-flow · commit c8cdedfb95dd · 2025-02-26T14:01:27.000-08:00
diff --git 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]:
