Skip to content

Commit e76ad23

Browse files
DarthMaxFlorentinD
authored andcommitted
Fix put command in arrow client
1 parent c483ce3 commit e76ad23

File tree

1 file changed

+3
-5
lines changed

1 file changed

+3
-5
lines changed

graphdatascience/query_runner/gds_arrow_client.py

Lines changed: 3 additions & 5 deletions
Original file line numberDiff line numberDiff line change
@@ -701,13 +701,11 @@ def _versioned_action_type(self, action_type: str) -> str:
701701

702702
def _versioned_flight_descriptor(self, flight_descriptor: dict[str, Any]) -> dict[str, Any]:
703703
return (
704-
flight_descriptor
705-
if self._arrow_endpoint_version == ArrowEndpointVersion.ALPHA
706-
else {
707-
"name": "PUT_MESSAGE",
704+
{
705+
"name": "PUT_COMMAND",
708706
"version": ArrowEndpointVersion.V1.version(),
709707
"body": flight_descriptor,
710-
}
708+
}
711709
)
712710

713711
@staticmethod

0 commit comments

Comments
 (0)