From 1104969f2f4f98df179628fbe7a008e9336f41b7 Mon Sep 17 00:00:00 2001 From: Thomas Cook Date: Fri, 24 Jan 2025 18:35:48 -0600 Subject: [PATCH] fix: spanner_cofig get logic and error when not defined --- graphistry/PlotterBase.py | 7 ++++--- 1 file changed, 4 insertions(+), 3 deletions(-) diff --git a/graphistry/PlotterBase.py b/graphistry/PlotterBase.py index f4ed0917a..885acfc8a 100644 --- a/graphistry/PlotterBase.py +++ b/graphistry/PlotterBase.py @@ -2530,13 +2530,14 @@ def spanner_gql_to_g(self: Plottable, query: str) -> Plottable: res = copy.copy(self) if not hasattr(res, '_spannergraph'): - spanner_config = PyGraphistry._config["spanner"] + spanner_config = PyGraphistry._config.get("spanner", None) + if spanner_config is not None: logger.debug(f"Spanner Config: {spanner_config}") else: - raise ValueError('spanner_config is None, use spanner_init() or register() passing spanner_config') + raise ValueError('spanner_config not defined. Pass spanner_config via register() and retry query.') - res = res.spanner_init(PyGraphistry._config["spanner"]) # type: ignore[attr-defined] + res = res.spanner_init(spanner_config) # type: ignore[attr-defined] return res._spannergraph.gql_to_graph(res, query)