Move files

great_ai/utilities/chunk.py

@@ -0,0 +1,17 @@
+from typing import Iterable, List, TypeVar
+
+T = TypeVar("T")
+
+
+def chunk(values: Iterable[T], chunk_size: int) -> Iterable[T]:
+    assert chunk_size >= 1
+
+    result: List[T] = []
+    for v in values:
+        result.append(v)
+        if len(result) == chunk_size:
+            yield result
+            result = []
+
+    if len(result) > 0:
+        yield result