From c9e4c5c7f7baae61841a444db76cf23933f1b226 Mon Sep 17 00:00:00 2001
From: Divyansh Khanna <divyanshkhanna@meta.com>
Date: Wed, 26 Feb 2025 09:58:00 -0800
Subject: [PATCH] fix_loader_state

---
 torchdata/nodes/loader.py | 2 +-
 1 file changed, 1 insertion(+), 1 deletion(-)

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]: