Explicit iterator shutdown (#1482)

nvski · web-flow · commit 10b87d6cb584 · 2025-05-06T10:48:38.000-07:00
diff --git a/torchdata/nodes/map.py b/torchdata/nodes/map.py
@@ -377,6 +377,7 @@ def __init__(
     def reset(self, initial_state: Optional[Dict[str, Any]] = None):
         super().reset(initial_state)
         if self._it is not None:
+            self._it._shutdown()
             del self._it
 
         if self.num_workers > 0:
