From d616989f57ead7d899a77b854ba4916f883bfdd6 Mon Sep 17 00:00:00 2001 From: Youri Wijnands Date: Mon, 10 Aug 2020 22:52:52 +0200 Subject: [PATCH] Update row_processor.py --- row_processor.py | 6 ++++-- 1 file changed, 4 insertions(+), 2 deletions(-) diff --git a/row_processor.py b/row_processor.py index 7aeb09c..195d2de 100644 --- a/row_processor.py +++ b/row_processor.py @@ -32,5 +32,7 @@ def batch(iterable, size): sourceiter = iter(iterable) while True: batchiter = islice(sourceiter, size) - yield chain([six.next(batchiter)], batchiter) - + try: + yield chain([six.next(batchiter)], batchiter) + except StopIteration: + return