Catch StopIteration errors for Python 3.7 compatibility (#132)

EntilZha · Sep 30, 2018 · 1feedea · 1feedea
1 parent 7a64bc5
commit 1feedea
Showing 1 changed file with 6 additions and 3 deletions.
diff --git a/functional/transformations.py b/functional/transformations.py
@@ -609,9 +609,12 @@ def grouped_impl(wrap, size, sequence):
     :return: grouped sequence
     """
     iterator = iter(sequence)
-    while True:
-        batch = islice(iterator, size)
-        yield list(chain((wrap(next(batch)),), batch))
+    try:
+        while True:
+            batch = islice(iterator, size)
+            yield list(chain((wrap(next(batch)),), batch))
+    except StopIteration:
+        return
 
 
 def grouped_t(wrap, size):