Remove unused arrow_client package

also format code

Author: FlorentinD

graphdatascience/__init__.py

@@ -1,4 +1,3 @@
-from .arrow_client.job_id import JobId
 from .graph.graph_create_result import GraphCreateResult
 from .graph.graph_object import Graph
 from .graph_data_science import GraphDataScience
@@ -34,5 +33,4 @@
     "NRModel",
     "GraphSageModel",
     "SimpleRelEmbeddingModel",
-    "JobId"
 ]

graphdatascience/arrow_client/job_id.py

@@ -700,13 +700,11 @@ def _versioned_action_type(self, action_type: str) -> str:
         return self._arrow_endpoint_version.prefix() + action_type
 
     def _versioned_flight_descriptor(self, flight_descriptor: dict[str, Any]) -> dict[str, Any]:
-        return (
-           {
-                "name": "PUT_COMMAND",
-                "version": ArrowEndpointVersion.V1.version(),
-                "body": flight_descriptor,
-           }
-        )
+        return {
+            "name": "PUT_COMMAND",
+            "version": ArrowEndpointVersion.V1.version(),
+            "body": flight_descriptor,
+        }
 
     @staticmethod
     def _sanitize_arrow_table(arrow_table: Table) -> Table:

graphdatascience/arrow_client/job_status.py

@@ -5,13 +5,15 @@
 from graphdatascience.query_runner.arrow_query_runner import ArrowQueryRunner
 from graphdatascience.server_version.server_version import ServerVersion
 
-from .conftest import CollectingQueryRunner
 from ...query_runner.arrow_endpoint_version import ArrowEndpointVersion
+from .conftest import CollectingQueryRunner
 
 
 @pytest.mark.parametrize("server_version", [ServerVersion(2, 6, 0)])
 def test_create(runner: CollectingQueryRunner) -> None:
-    arrow_info = ArrowInfo(listenAddress="localhost:1234", enabled=True, running=True, versions=[ArrowEndpointVersion.V1.version()])
+    arrow_info = ArrowInfo(
+        listenAddress="localhost:1234", enabled=True, running=True, versions=[ArrowEndpointVersion.V1.version()]
+    )
     arrow_runner = ArrowQueryRunner.create(runner, arrow_info)
 
     assert isinstance(arrow_runner, ArrowQueryRunner)
@@ -30,7 +32,9 @@ def test_return_fallback_when_arrow_is_not_enabled(runner: CollectingQueryRunner
 
 @pytest.mark.parametrize("server_version", [ServerVersion(2, 6, 0)])
 def test_create_with_provided_connection(runner: CollectingQueryRunner) -> None:
-    arrow_info = ArrowInfo(listenAddress="localhost:1234", enabled=True, running=True, versions=[ArrowEndpointVersion.V1.version()])
+    arrow_info = ArrowInfo(
+        listenAddress="localhost:1234", enabled=True, running=True, versions=[ArrowEndpointVersion.V1.version()]
+    )
 
     arrow_runner = ArrowQueryRunner.create(runner, arrow_info, connection_string_override="localhost:4321")