Update subgraphx.rst

Add a missing import of Data
divelab · Feb 27, 2023 · 2728cda · 2728cda
1 parent 55c40a7
commit 2728cda
Showing 1 changed file with 1 addition and 0 deletions.
diff --git a/docs/source/tutorials/subgraphx.rst b/docs/source/tutorials/subgraphx.rst
@@ -82,6 +82,7 @@ Since the graph model is a two-layer GNN model, the information only aggregates
     from dig.xgraph.method.subgraphx import PlotUtils
     from dig.xgraph.method.subgraphx import MCTS
     from torch_geometric.utils import to_networkx
+    from torch_geometric.data import Data
 
     subgraph_x, subgraph_edge_index, subset, edge_mask, kwargs = \
         MCTS.__subgraph__(node_idx, data.x, data.edge_index, num_hops=2)