feat : Added new generator - batched_iterable.

Author: manas-shinde

generators/batched_iterable.py

@@ -0,0 +1,10 @@
+def batched_iterable(iterable, batch_size: int):
+    """Yields items from iterable in batches of `batch_size`."""
+    batch = []
+    for item in iterable:
+        batch.append(item)
+        if len(batch) == batch_size:
+            yield batch
+            batch = []
+    if batch:
+        yield batch

tests/test_batched_iterable.py

@@ -0,0 +1,7 @@
+from generators.batched_iterable import batched_iterable
+
+
+def test_batches():
+    data = list(range(10))
+    batches = list(batched_iterable(data, 4))
+    assert batches == [[0, 1, 2, 3], [4, 5, 6, 7], [8, 9]]