Skip to content

Commit 98d3357

Browse files
committed
Avoid showing irrelevant warning for aura users
1 parent 0fad11b commit 98d3357

File tree

2 files changed

+8
-1
lines changed

2 files changed

+8
-1
lines changed

changelog/1.2.0.md

Lines changed: 1 addition & 0 deletions
Original file line numberDiff line numberDiff line change
@@ -10,6 +10,7 @@
1010
## Bug fixes
1111

1212
* Fixed a bug where the `separate_property_columns=True` option of `gds.graph.streamNodeProperties` did not handle list node properties correctly.
13+
* Fixed a bug where an irrelevant warning was shown when creating a `GraphDataScience` object targeting an AuraDS instance with GDS server version >= 2.1.0.
1314

1415

1516
## Improvements

graphdatascience/graph_data_science.py

Lines changed: 7 additions & 1 deletion
Original file line numberDiff line numberDiff line change
@@ -80,7 +80,13 @@ def __init__(
8080
arrow_info["listenAddress"], self._query_runner, auth, driver.encrypted, True
8181
)
8282
except Exception as e:
83-
warnings.warn(f"Could not initialize GDS Flight Server client: {e}")
83+
# AuraDS does not have arrow support at this time, so we should not warn about it.
84+
# TODO: Remove this check when AuraDS gets arrow support.
85+
if (
86+
"There is no procedure with the name `gds.debug.arrow` "
87+
"registered for this database instance." not in str(e)
88+
):
89+
warnings.warn(f"Could not initialize GDS Flight Server client: {e}")
8490

8591
super().__init__(self._query_runner, "gds", self._server_version)
8692

0 commit comments

Comments
 (0)