diff --git a/graphistry/Plottable.py b/graphistry/Plottable.py index 9f10f42f6..21aaea830 100644 --- a/graphistry/Plottable.py +++ b/graphistry/Plottable.py @@ -62,7 +62,8 @@ class Plottable(object): _complex_encodings : dict _bolt_driver : Any _tigergraph : Any - + _spannergraph: SpannerGraph = None + _dataset_id: Optional[str] _url: Optional[str] _nodes_file_id: Optional[str] diff --git a/graphistry/PlotterBase.py b/graphistry/PlotterBase.py index f9b9367c4..ac38cd5eb 100644 --- a/graphistry/PlotterBase.py +++ b/graphistry/PlotterBase.py @@ -178,7 +178,7 @@ def __init__(self, *args: Any, **kwargs: Any) -> None: # Integrations self._bolt_driver : Any = None self._tigergraph : Any = None - self._spannergraph : Any = None + self._spannergraph: SpannerGraph = None # feature engineering self._node_embedding = None @@ -2546,7 +2546,7 @@ def spanner_gql_to_g(self: Plottable, query: str) -> Plottable: else: logger.warning('PyGraphistry._config["spanner"] is None') - res = res.spanner_init(PyGraphistry._config["spanner"]) + res = res.spanner_init(PyGraphistry._config["spanner"]) # type: ignore[attr-defined] return res._spannergraph.gql_to_graph(query)