Fix Python 3.7 compatibility issue

Catch StopIteration when grouping records with padding
computationalprivacy · May 18, 2020 · d02ffe8 · d02ffe8
1 parent fe97be9
commit d02ffe8
Showing 1 changed file with 4 additions and 1 deletion.
diff --git a/bandicoot/helper/group.py b/bandicoot/helper/group.py
@@ -180,7 +180,10 @@ def group_records_with_padding(records, groupby='week'):
 
         yield chunk
 
-        pointer = next(_range)
+        try:
+            pointer = next(_range)
+        except StopIteration as e:
+            pass
 
 
 def group_records(records, groupby='week'):