Skip to content

Commit c6e75ef

Browse files
committed
Improve typing
(cherry picked from commit af5bc45)
1 parent b2aa537 commit c6e75ef

File tree

2 files changed

+4
-2
lines changed

2 files changed

+4
-2
lines changed

graphdatascience/query_runner/standalone_session_query_runner.py

Lines changed: 3 additions & 1 deletion
Original file line numberDiff line numberDiff line change
@@ -1,3 +1,5 @@
1+
from __future__ import annotations
2+
13
from typing import Any, Optional
24

35
from pandas import DataFrame
@@ -75,5 +77,5 @@ def last_bookmarks(self) -> Optional[Any]:
7577
def set_server_version(self, _: ServerVersion) -> None:
7678
super().set_server_version(_)
7779

78-
def clone(self, host: str, port: int) -> "QueryRunner":
80+
def clone(self, host: str, port: int) -> QueryRunner:
7981
return self

graphdatascience/tests/unit/conftest.py

Lines changed: 1 addition & 1 deletion
Original file line numberDiff line numberDiff line change
@@ -126,7 +126,7 @@ def create_graph_constructor(
126126
self, graph_name, concurrency, undirected_relationship_types, self._server_version
127127
)
128128

129-
def clone(self, host: str, port: int) -> "QueryRunner":
129+
def clone(self, host: str, port: int) -> QueryRunner:
130130
return self
131131

132132
def set__mock_result(self, result: DataFrame) -> None:

0 commit comments

Comments
 (0)