Skip to content

Commit ded9752

Browse files
committed
Merge branch 'pr/15'
2 parents 5e45aba + 94ff8dd commit ded9752

File tree

1 file changed

+12
-8
lines changed

1 file changed

+12
-8
lines changed

progress/__init__.py

Lines changed: 12 additions & 8 deletions
Original file line numberDiff line numberDiff line change
@@ -73,10 +73,12 @@ def next(self, n=1):
7373
self.update()
7474

7575
def iter(self, it):
76-
for x in it:
77-
yield x
78-
self.next()
79-
self.finish()
76+
try:
77+
for x in it:
78+
yield x
79+
self.next()
80+
finally:
81+
self.finish()
8082

8183

8284
class Progress(Infinite):
@@ -117,7 +119,9 @@ def iter(self, it):
117119
except TypeError:
118120
pass
119121

120-
for x in it:
121-
yield x
122-
self.next()
123-
self.finish()
122+
try:
123+
for x in it:
124+
yield x
125+
self.next()
126+
finally:
127+
self.finish()

0 commit comments

Comments
 (0)