Skip to content

Commit 1104969

Browse files
committed
fix: spanner_cofig get logic and error when not defined
1 parent 52e9b4b commit 1104969

File tree

1 file changed

+4
-3
lines changed

1 file changed

+4
-3
lines changed

graphistry/PlotterBase.py

Lines changed: 4 additions & 3 deletions
Original file line numberDiff line numberDiff line change
@@ -2530,13 +2530,14 @@ def spanner_gql_to_g(self: Plottable, query: str) -> Plottable:
25302530
res = copy.copy(self)
25312531

25322532
if not hasattr(res, '_spannergraph'):
2533-
spanner_config = PyGraphistry._config["spanner"]
2533+
spanner_config = PyGraphistry._config.get("spanner", None)
2534+
25342535
if spanner_config is not None:
25352536
logger.debug(f"Spanner Config: {spanner_config}")
25362537
else:
2537-
raise ValueError('spanner_config is None, use spanner_init() or register() passing spanner_config')
2538+
raise ValueError('spanner_config not defined. Pass spanner_config via register() and retry query.')
25382539

2539-
res = res.spanner_init(PyGraphistry._config["spanner"]) # type: ignore[attr-defined]
2540+
res = res.spanner_init(spanner_config) # type: ignore[attr-defined]
25402541

25412542
return res._spannergraph.gql_to_graph(res, query)
25422543

0 commit comments

Comments
 (0)