Fix issue #131

2018-11-07 15:34:52 +01:00 · 2018-11-07 15:34:52 +01:00 · d87f2ec37d
commit d87f2ec37d
parent 4ef32a5653
1 changed files with 4 additions and 1 deletions
--- a/flatisfy/tools.py
+++ b/flatisfy/tools.py
@ -169,7 +169,10 @@ def batch(iterable, size):
    sourceiter = iter(iterable)
    while True:
        batchiter = itertools.islice(sourceiter, size)
-        yield itertools.chain([next(batchiter)], batchiter)
+        try:
+            yield itertools.chain([next(batchiter)], batchiter)
+        except StopIteration:
+            return


 def is_within_interval(value, min_value=None, max_value=None):