Fix wrong arguments.

divelab · Feb 27, 2023 · 7ed1646 · 7ed1646
1 parent 2728cda
commit 7ed1646
Showing 1 changed file with 1 addition and 2 deletions.
diff --git a/docs/source/tutorials/subgraphx.rst b/docs/source/tutorials/subgraphx.rst
@@ -131,8 +131,7 @@ After MCTS searching and Shapley value computation, the subgraph with the highes
     result = find_closest_node_result(explanation_results[prediction], max_nodes=max_nodes)
 
     plotutils = PlotUtils(dataset_name='ba_shapes')
-    explainer.visualization(explanation_results,
-                            prediction,
+    explainer.visualization(explanation_results[prediction],
                             max_nodes=max_nodes,
                             plot_utils=plotutils,
                             y=data.y)