Missing files

Signed-off-by: Alexey Kudinkin <ak@anyscale.com>

python/ray/_private/collections.py

@@ -0,0 +1,10 @@
+from typing import List, Any
+
+
+def split(items: List[Any], chunk_size: int):
+    """Splits provided list into chunks of given size"""
+
+    assert chunk_size > 0, "Chunk size has to be > 0"
+
+    for i in range(0, len(items), chunk_size):
+        yield items[i : i + chunk_size]

python/ray/tests/test_collections.py

@@ -0,0 +1,18 @@
+import sys
+
+import pytest
+
+from ray._private.collections import split
+
+
+def test_split():
+    ints = list(range(0, 5))
+
+    assert list(split(ints, 5)) == [ints]
+    assert list(split(ints, 10)) == [ints]
+    assert list(split(ints, 1)) == [[e] for e in ints]
+    assert list(split(ints, 2)) == [[0, 1], [2, 3], [4]]
+
+
+if __name__ == "__main__":
+    sys.exit(pytest.main(["-v", __file__]))