Fix formatting

Author: DarthMax

graphdatascience/tests/integrationV2/procedure_surface/arrow/catalog/test_node_label_arrow_endpoints.py

@@ -1,4 +1,3 @@
-import json
 from typing import Generator
 
 import pytest

graphdatascience/tests/integrationV2/procedure_surface/arrow/graph_creation_helper.py

@@ -1,6 +1,5 @@
 from contextlib import contextmanager
-from types import TracebackType
-from typing import Optional, Tuple, Type, Any, Generator
+from typing import Any, Generator, Optional, Tuple
 
 from graphdatascience import Graph, QueryRunner
 from graphdatascience.arrow_client.authenticated_flight_client import AuthenticatedArrowClient
@@ -16,6 +15,7 @@ def __init__(self, name: str):
     def name(self) -> str:
         return self._name
 
+
 @contextmanager
 def create_graph(
     arrow_client: AuthenticatedArrowClient, graph_name: str, gdl: str, undirected: Optional[Tuple[str, str]] = None
@@ -41,6 +41,7 @@ def create_graph(
     finally:
         CatalogArrowEndpoints(arrow_client).drop(graph_name, fail_if_missing=False)
 
+
 @contextmanager
 def create_graph_from_db(
     arrow_client: AuthenticatedArrowClient,
@@ -50,7 +51,7 @@ def create_graph_from_db(
     query: str,
 ) -> Generator[TestArrowGraph, Any, None]:
     try:
-        query_runner.run_cypher(graph_data);
+        query_runner.run_cypher(graph_data)
         CatalogArrowEndpoints(arrow_client, query_runner).project(
             graph_name=graph_name,
             query=query,
@@ -59,4 +60,4 @@ def create_graph_from_db(
         yield TestArrowGraph(graph_name)
     finally:
         CatalogArrowEndpoints(arrow_client).drop(graph_name, fail_if_missing=False)
-        query_runner.run_cypher("MATCH (n) DETACH DELETE n")
+        query_runner.run_cypher("MATCH (n) DETACH DELETE n")

graphdatascience/tests/integrationV2/procedure_surface/arrow/test_articlerank_arrow_endpoints.py

@@ -1,4 +1,3 @@
-import json
 from typing import Generator
 
 import pytest

graphdatascience/tests/integrationV2/procedure_surface/arrow/test_articulationpoints_arrow_endpoints.py

@@ -1,4 +1,3 @@
-import json
 from typing import Generator
 
 import pytest

graphdatascience/tests/integrationV2/procedure_surface/arrow/test_betweenness_arrow_endpoints.py

@@ -1,4 +1,3 @@
-import json
 from typing import Generator
 
 import pytest

graphdatascience/tests/integrationV2/procedure_surface/arrow/test_celf_arrow_endpoints.py

@@ -1,4 +1,3 @@
-import json
 from typing import Generator
 
 import pytest

graphdatascience/tests/integrationV2/procedure_surface/arrow/test_closeness_arrow_endpoints.py

@@ -1,4 +1,3 @@
-import json
 from typing import Generator
 
 import pytest

graphdatascience/tests/integrationV2/procedure_surface/arrow/test_closeness_harmonic_arrow_endpoints.py

@@ -1,4 +1,3 @@
-import json
 from typing import Generator
 
 import pytest

graphdatascience/tests/integrationV2/procedure_surface/arrow/test_degree_arrow_endpoints.py

@@ -1,4 +1,3 @@
-import json
 from typing import Generator
 
 import pytest

graphdatascience/tests/integrationV2/procedure_surface/arrow/test_eigenvector_arrow_endpoints.py

@@ -1,4 +1,3 @@
-import json
 from typing import Generator
 
 import pytest