Expose arrow configuration as a remote projection parameter

Co-authored-by: Max Kießling <max.kiessling@neotechnology.com>

Author: Mats-SX

graphdatascience/graph/graph_remote_project_runner.py

@@ -20,6 +20,7 @@ def __call__(
         concurrency: int = 4,
         undirected_relationship_types: Optional[List[str]] = None,
         inverse_indexed_relationship_types: Optional[List[str]] = None,
+        batch_size: Optional[int] = None,
     ) -> GraphCreateResult:
         if inverse_indexed_relationship_types is None:
             inverse_indexed_relationship_types = []
@@ -32,6 +33,7 @@ def __call__(
             concurrency=concurrency,
             undirected_relationship_types=undirected_relationship_types,
             inverse_indexed_relationship_types=inverse_indexed_relationship_types,
+            arrow_configuration={"batchSize": batch_size},
         )
 
         result = self._query_runner.call_procedure(

graphdatascience/tests/integration/test_remote_graph_ops.py

@@ -43,6 +43,17 @@ def test_remote_projection(gds_with_cloud_setup: AuraGraphDataScience) -> None:
     assert result["nodeCount"] == 3
 
 
+@pytest.mark.cloud_architecture
+@pytest.mark.compatible_with(min_inclusive=ServerVersion(2, 7, 0))
+def test_remote_projection_with_small_batch_size(gds_with_cloud_setup: AuraGraphDataScience) -> None:
+    G, result = gds_with_cloud_setup.graph.project(
+        GRAPH_NAME, "MATCH (n)-->(m) RETURN gds.graph.project.remote(n, m)", batch_size=10
+    )
+
+    assert G.name() == GRAPH_NAME
+    assert result["nodeCount"] == 3
+
+
 @pytest.mark.cloud_architecture
 @pytest.mark.compatible_with(min_inclusive=ServerVersion(2, 7, 0))
 def test_remote_write_back_page_rank(gds_with_cloud_setup: AuraGraphDataScience) -> None: