|
3 | 3 |
|
4 | 4 | from pandas import DataFrame |
5 | 5 |
|
6 | | -from ..call_parameters import CallParameters |
7 | | -from .gds_arrow_client import GdsArrowClient |
8 | | -from .query_runner import QueryRunner |
9 | 6 | from graphdatascience.query_runner.graph_constructor import GraphConstructor |
10 | 7 | from graphdatascience.server_version.server_version import ServerVersion |
| 8 | +from .gds_arrow_client import GdsArrowClient |
| 9 | +from .query_runner import QueryRunner |
| 10 | +from ..call_parameters import CallParameters |
11 | 11 |
|
12 | 12 |
|
13 | 13 | class AuraDbQueryRunner(QueryRunner): |
@@ -47,7 +47,7 @@ def call_procedure( |
47 | 47 | params = CallParameters() |
48 | 48 |
|
49 | 49 | if AuraDbQueryRunner.GDS_REMOTE_PROJECTION_PROC_NAME == endpoint: |
50 | | - return self._remote_projection(endpoint, params, yields, database, logging, custom_error) |
| 50 | + return self._remote_projection(endpoint, params, yields, database, logging) |
51 | 51 |
|
52 | 52 | elif ".write" in endpoint and self.is_remote_projected_graph(params["graph_name"]): |
53 | 53 | return self._remote_write_back(endpoint, params, yields, database, logging, custom_error) |
@@ -103,7 +103,6 @@ def _remote_projection( |
103 | 103 | yields: Optional[List[str]] = None, |
104 | 104 | database: Optional[str] = None, |
105 | 105 | logging: bool = False, |
106 | | - custom_error: bool = True, |
107 | 106 | ) -> DataFrame: |
108 | 107 | self._inject_connection_parameters(params) |
109 | 108 | return self._db_query_runner.call_procedure(endpoint, params, yields, database, logging, False) |
|
0 commit comments