Refactor: Simplify iterator method in HDF5DataStore

ccpem · Oct 9, 2024 · b24d299 · b24d299
1 parent 36a5883
commit b24d299
Showing 1 changed file with 1 addition and 2 deletions.
diff --git a/src/caked/hdf5.py b/src/caked/hdf5.py
@@ -69,8 +69,7 @@ def __getitem__(self, key: str):
 
     def __iter__(self):
         with h5py.File(self.save_path, "r") as f:
-            for key in f:
-                yield key
+            yield from f
 
     def get(self, key: str, default=None, to_torch: bool = False):
         try: