Skip to content

Commit 65d9b93

Browse files
committed
Make sure to close the projection query runner after remote projection is done
* also deletes unused code (cherry picked from commit 402e658)
1 parent 7dbb8e9 commit 65d9b93

File tree

2 files changed

+5
-3
lines changed

2 files changed

+5
-3
lines changed

graphdatascience/query_runner/neo4j_query_runner.py

Lines changed: 0 additions & 2 deletions
Original file line numberDiff line numberDiff line change
@@ -145,8 +145,6 @@ def run_cypher(
145145
connectivity_retry_config = Neo4jQueryRunner.ConnectivityRetriesConfig()
146146
self._verify_connectivity(database=database, retry_config=connectivity_retry_config)
147147

148-
self._driver.session()
149-
150148
with self._driver.session(database=database, bookmarks=self.bookmarks()) as session:
151149
try:
152150
result = session.run(query, params)

graphdatascience/query_runner/protocol/project_protocols.py

Lines changed: 5 additions & 1 deletion
Original file line numberDiff line numberDiff line change
@@ -150,4 +150,8 @@ def project_fn() -> DataFrame:
150150
ProtocolVersion.V3.versioned_procedure_name(endpoint), params, yields, database, logging, False
151151
)
152152

153-
return project_fn()
153+
projection_result = project_fn()
154+
155+
projection_query_runner.close()
156+
157+
return projection_result

0 commit comments

Comments
 (0)