Fix typed examples iterable state dict (#7121)

fix typed examples iterable state dict
huggingface · Aug 22, 2024 · 3813ce8 · 3813ce8
1 parent cedffa5
commit 3813ce8
Showing 1 changed file with 1 addition and 2 deletions.
diff --git a/src/datasets/iterable_dataset.py b/src/datasets/iterable_dataset.py
@@ -1601,8 +1601,7 @@ def iter_arrow(self):
             return self._iter_arrow
 
     def _init_state_dict(self) -> dict:
-        if not self._state_dict:
-            self._state_dict = self.ex_iterable._init_state_dict()
+        self._state_dict = self.ex_iterable._init_state_dict()
         return self._state_dict
 
     def __iter__(self):