fix LMDBSerializer

tensorpack · Jan 9, 2020 · ffd9eae · ffd9eae
1 parent bf8acbf
commit ffd9eae
Showing 1 changed file with 1 addition and 2 deletions.
diff --git a/tensorpack/dataflow/format.py b/tensorpack/dataflow/format.py
@@ -145,8 +145,7 @@ def __iter__(self):
         with self._guard:
             if not self._shuffle:
                 c = self._txn.cursor()
-                while next(c):
-                    k, v = c.item()
+                for k, v in c:
                     if k != b'__keys__':
                         yield [k, v]
             else: