Fix formatting

Author: DarthMax

graphdatascience/arrow_client/middleware/AuthMiddleware.py

@@ -2,7 +2,7 @@
 
 import base64
 import time
-from typing import Optional, Any
+from typing import Any, Optional
 
 from pyarrow._flight import ClientMiddleware, ClientMiddlewareFactory
 
@@ -60,4 +60,4 @@ def sending_headers(self) -> dict[str, str]:
         auth_token = f"{auth_pair[0]}:{auth_pair[1]}"
         auth_token = "Basic " + base64.b64encode(auth_token.encode("utf-8")).decode("ASCII")
         # There seems to be a bug, `authorization` must be lower key
-        return {"authorization": auth_token}
+        return {"authorization": auth_token}

graphdatascience/arrow_client/middleware/UserAgentMiddleware.py

@@ -4,6 +4,7 @@
 
 from pyarrow._flight import ClientMiddleware, ClientMiddlewareFactory
 
+
 class UserAgentFactory(ClientMiddlewareFactory):  # type: ignore
     def __init__(self, useragent: str, *args: Any, **kwargs: Any) -> None:
         super().__init__(*args, **kwargs)
@@ -12,6 +13,7 @@ def __init__(self, useragent: str, *args: Any, **kwargs: Any) -> None:
     def start_call(self, info: Any) -> ClientMiddleware:
         return self._middleware
 
+
 class UserAgentMiddleware(ClientMiddleware):  # type: ignore
     def __init__(self, useragent: str, *args: Any, **kwargs: Any) -> None:
         super().__init__(*args, **kwargs)
@@ -21,4 +23,4 @@ def sending_headers(self) -> dict[str, str]:
         return {"x-gds-user-agent": self._useragent}
 
     def received_headers(self, headers: dict[str, Any]) -> None:
-        pass
+        pass

graphdatascience/arrow_client/v2/api_types.py

@@ -1,9 +1,11 @@
 from dataclasses import dataclass
 
+
 @dataclass(frozen=True, repr=True)
 class JobIdConfig:
     jobId: str
 
+
 @dataclass(frozen=True, repr=True)
 class JobStatus:
     jobId: str

graphdatascience/graph_data_science.py

@@ -9,14 +9,14 @@
 from pandas import DataFrame
 
 from graphdatascience.arrow_client.arrow_authentication import UsernamePasswordAuthentication
+from graphdatascience.arrow_client.arrow_info import ArrowInfo
 from graphdatascience.procedure_surface.api.wcc_endpoints import WccEndpoints
 from graphdatascience.procedure_surface.cypher.wcc_proc_runner import WccCypherEndpoints
 
 from .call_builder import IndirectCallBuilder
 from .endpoints import AlphaEndpoints, BetaEndpoints, DirectEndpoints
 from .error.uncallable_namespace import UncallableNamespace
 from .graph.graph_proc_runner import GraphProcRunner
-from graphdatascience.arrow_client.arrow_info import ArrowInfo
 from .query_runner.arrow_query_runner import ArrowQueryRunner
 from .query_runner.neo4j_query_runner import Neo4jQueryRunner
 from .query_runner.query_runner import QueryRunner

graphdatascience/query_runner/arrow_query_runner.py

@@ -6,11 +6,11 @@
 from pandas import DataFrame
 
 from graphdatascience.arrow_client.arrow_authentication import ArrowAuthentication
+from graphdatascience.arrow_client.arrow_info import ArrowInfo
 from graphdatascience.query_runner.query_mode import QueryMode
 from graphdatascience.retry_utils.retry_config import RetryConfig
 
 from ..call_parameters import CallParameters
-from graphdatascience.arrow_client.arrow_info import ArrowInfo
 from ..server_version.server_version import ServerVersion
 from .arrow_graph_constructor import ArrowGraphConstructor
 from .gds_arrow_client import GdsArrowClient

graphdatascience/query_runner/gds_arrow_client.py

@@ -36,13 +36,13 @@
 )
 
 from graphdatascience.arrow_client.arrow_authentication import ArrowAuthentication, UsernamePasswordAuthentication
+from graphdatascience.arrow_client.arrow_info import ArrowInfo
 from graphdatascience.retry_utils.retry_config import RetryConfig
 from graphdatascience.retry_utils.retry_utils import before_log
 
 from ..semantic_version.semantic_version import SemanticVersion
 from ..version import __version__
 from .arrow_endpoint_version import ArrowEndpointVersion
-from graphdatascience.arrow_client.arrow_info import ArrowInfo
 
 
 class GdsArrowClient:

graphdatascience/query_runner/protocol/project_protocols.py

@@ -25,7 +25,7 @@ def project_params(
     def run_projection(
         self,
         query_runner: QueryRunner,
-        endpoint: str,
+        query: str,
         params: CallParameters,
         terminationFlag: TerminationFlag,
         yields: Optional[list[str]] = None,

graphdatascience/session/aura_graph_data_science.py

@@ -5,6 +5,8 @@
 from pandas import DataFrame
 
 from graphdatascience import QueryRunner, ServerVersion
+from graphdatascience.arrow_client.arrow_authentication import ArrowAuthentication
+from graphdatascience.arrow_client.arrow_info import ArrowInfo
 from graphdatascience.call_builder import IndirectCallBuilder
 from graphdatascience.endpoints import (
     AlphaRemoteEndpoints,
@@ -13,8 +15,6 @@
 )
 from graphdatascience.error.uncallable_namespace import UncallableNamespace
 from graphdatascience.graph.graph_remote_proc_runner import GraphRemoteProcRunner
-from graphdatascience.arrow_client.arrow_authentication import ArrowAuthentication
-from graphdatascience.arrow_client.arrow_info import ArrowInfo
 from graphdatascience.query_runner.arrow_query_runner import ArrowQueryRunner
 from graphdatascience.query_runner.gds_arrow_client import GdsArrowClient
 from graphdatascience.query_runner.neo4j_query_runner import Neo4jQueryRunner

graphdatascience/tests/integration/conftest.py

@@ -6,8 +6,8 @@
 import pytest
 from neo4j import Driver, GraphDatabase
 
-from graphdatascience.graph_data_science import GraphDataScience
 from graphdatascience.arrow_client.arrow_authentication import UsernamePasswordAuthentication
+from graphdatascience.graph_data_science import GraphDataScience
 from graphdatascience.query_runner.neo4j_query_runner import Neo4jQueryRunner
 from graphdatascience.server_version.server_version import ServerVersion
 from graphdatascience.session.aura_graph_data_science import AuraGraphDataScience

graphdatascience/tests/unit/arrow_client/test_data_mapper.py

@@ -1,5 +1,5 @@
 from dataclasses import dataclass
-from typing import Dict, Any
+from typing import Any, Dict
 
 import pytest
 
@@ -22,12 +22,8 @@ class ExampleDataclass:
     "input_data, expected_output",
     [
         (
-                {
-                    "field_one": "test",
-                    "field_two": 123,
-                    "nested": {"nested_field": 456}
-                },
-                ExampleDataclass("test", 123, NestedDataclass(456)),
+            {"field_one": "test", "field_two": 123, "nested": {"nested_field": 456}},
+            ExampleDataclass("test", 123, NestedDataclass(456)),
         ),
     ],
 )
@@ -37,12 +33,7 @@ def test_dict_to_dataclass(input_data: Dict[str, Any], expected_output: ExampleD
 
 
 def test_dict_to_dataclass_strict_mode_rejects_extra_fields():
-    input_data = {
-        "field_one": "test",
-        "field_two": 123,
-        "nested": {"nested_field": 456},
-        "extra_field": "not_allowed"
-    }
+    input_data = {"field_one": "test", "field_two": 123, "nested": {"nested_field": 456}, "extra_field": "not_allowed"}
 
     with pytest.raises(ValueError, match="Extra field 'extra_field' not allowed in ExampleDataclass"):
         DataMapper.dict_to_dataclass(input_data, ExampleDataclass, strict=True)