Skip to content

Commit 26cc360

Browse files
DarthMaxMats-SX
authored andcommitted
Do not set the batch size if it is not defined by the user
1 parent fe6e2f3 commit 26cc360

File tree

1 file changed

+5
-1
lines changed

1 file changed

+5
-1
lines changed

graphdatascience/graph/graph_remote_project_runner.py

Lines changed: 5 additions & 1 deletion
Original file line numberDiff line numberDiff line change
@@ -27,13 +27,17 @@ def __call__(
2727
if undirected_relationship_types is None:
2828
undirected_relationship_types = []
2929

30+
arrow_configuration = {}
31+
if batch_size is not None:
32+
arrow_configuration["batchSize"] = batch_size
33+
3034
params = CallParameters(
3135
graph_name=graph_name,
3236
query=query,
3337
concurrency=concurrency,
3438
undirected_relationship_types=undirected_relationship_types,
3539
inverse_indexed_relationship_types=inverse_indexed_relationship_types,
36-
arrow_configuration={"batchSize": batch_size},
40+
arrow_configuration=arrow_configuration,
3741
)
3842

3943
result = self._query_runner.call_procedure(

0 commit comments

Comments
 (0)