fix 261 (#262)

9588d6ca · Patrick Wieschollek · Yuxin Wu · 0ada5796 · 9588d6ca
Commit 9588d6ca authored May 09, 2017 by Patrick Wieschollek Committed by Yuxin Wu May 09, 2017
Hide whitespace changes
Inline Side-by-side

Showing with 2 additions and 2 deletions

tensorpack/dataflow/common.py tensorpack/dataflow/common.py +2 -2

No files found.
--- a/tensorpack/dataflow/common.py
+++ b/tensorpack/dataflow/common.py
@@ -198,10 +198,10 @@ class FixedSizeData(ProxyDataFlow):
        cnt = 0
        while True:
            try:
-                dp = self.itr.next()
+                dp = next(self.itr)
            except StopIteration:
                self.itr = self.ds.get_data()
-                dp = self.itr.next()
+                dp = next(self.itr)

            cnt += 1
            yield dp