Apply codestyle

(cherry picked from commit 6402e70)

Author: DarthMax

graphdatascience/query_runner/neo4j_query_runner.py

@@ -295,10 +295,9 @@ def clone(self, endpoint: str) -> QueryRunner:
             self._auto_close,
             self._bookmarks,
             self._show_progress,
-            self._instance_description
+            self._instance_description,
         )
 
-
     @staticmethod
     def handle_driver_exception(session: neo4j.Session, e: Exception) -> None:
         reg_gds_hit = re.search(

graphdatascience/query_runner/protocol/project_protocols.py

@@ -2,13 +2,11 @@
 from logging import DEBUG, getLogger
 from typing import Any, Optional
 
-import neo4j
 from pandas import DataFrame
 from tenacity import retry, retry_if_result, wait_incrementing
 
 from graphdatascience import QueryRunner
 from graphdatascience.call_parameters import CallParameters
-from graphdatascience.query_runner.neo4j_query_runner import Neo4jQueryRunner
 from graphdatascience.query_runner.protocol.status import Status
 from graphdatascience.query_runner.termination_flag import TerminationFlag
 from graphdatascience.retry_utils.retry_utils import before_log

graphdatascience/query_runner/query_runner.py

@@ -81,7 +81,7 @@ def set_show_progress(self, show_progress: bool) -> None:
         pass
 
     @abstractmethod
-    def clone(self, endpoint:str) -> "QueryRunner":
+    def clone(self, endpoint: str) -> "QueryRunner":
         pass
 
     def set_server_version(self, _: ServerVersion) -> None:

graphdatascience/query_runner/session_query_runner.py

@@ -125,7 +125,7 @@ def clone(self, endpoint: str) -> QueryRunner:
             self._gds_query_runner.clone(endpoint),
             self._db_query_runner.clone(endpoint),
             self._gds_arrow_client,
-            self._show_progress
+            self._show_progress,
         )
 
     def close(self) -> None:
@@ -153,6 +153,7 @@ def _remote_projection(
         project_params = project_protocol.project_params(graph_name, query, job_id, params, arrow_config)
 
         try:
+
             def run_projection() -> DataFrame:
                 return project_protocol.run_projection(
                     self._db_query_runner, endpoint, project_params, terminationFlag, yields, database, logging

graphdatascience/query_runner/standalone_session_query_runner.py

@@ -76,6 +76,4 @@ def set_server_version(self, _: ServerVersion) -> None:
         super().set_server_version(_)
 
     def clone(self, endpoint: str) -> "QueryRunner":
-        return StandaloneSessionQueryRunner(
-            self._query_runner.clone(endpoint)
-        )
+        return StandaloneSessionQueryRunner(self._query_runner.clone(endpoint))