Update row_processor.py (Networks-Learning#11)

Youri Wijnands · web-flow · commit 491e552cfcd7 · 2020-08-11T13:36:35.000+02:00
diff --git 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
