Merge pull request #657 from FlorentinD/push-based-proj-auth-problem

Use session credentials for GdsArrowClient
neo4j · Jun 6, 2024 · 972680d · 972680d
2 parents ead1729 + 67eb503
commit 972680d
Showing 1 changed file with 5 additions and 1 deletion.
diff --git a/graphdatascience/session/aura_graph_data_science.py b/graphdatascience/session/aura_graph_data_science.py
@@ -63,7 +63,11 @@ def __init__(
         self._db_query_runner.set_database("neo4j")
 
         arrow_client = GdsArrowClient.create(
-            gds_query_runner, aura_db_connection_info.auth(), self._db_query_runner.encrypted()
+            gds_neo4j_query_runner,
+            gds_session_connection_info.auth(),
+            gds_neo4j_query_runner.encrypted(),
+            arrow_disable_server_verification,
+            arrow_tls_root_certs,
         )
         self._query_runner = AuraDbQueryRunner(
             gds_query_runner, self._db_query_runner, arrow_client, self._db_query_runner.encrypted()