Test CaptumExplainer and GNNExplainer with unused parameters (#9216)

pyg-team · Apr 19, 2024 · ed17034 · ed17034
1 parent 557d063
commit ed17034
Show file tree

Hide file tree

Showing 2 changed files with 6 additions and 0 deletions.
diff --git a/test/explain/algorithm/test_captum_explainer.py b/test/explain/algorithm/test_captum_explainer.py
@@ -48,6 +48,9 @@ def __init__(self, model_config: ModelConfig):
         self.conv1 = GCNConv(3, 16)
         self.conv2 = GCNConv(16, out_channels)
 
+        # Add unused parameter:
+        self.param = torch.nn.Parameter(torch.empty(1))
+
     def forward(self, x, edge_index, batch=None, edge_label_index=None):
         x = self.conv1(x, edge_index).relu()
         x = self.conv2(x, edge_index)

diff --git a/test/explain/algorithm/test_gnn_explainer.py b/test/explain/algorithm/test_gnn_explainer.py
@@ -26,6 +26,9 @@ def __init__(self, model_config: ModelConfig):
         self.conv1 = GCNConv(3, 16)
         self.conv2 = GCNConv(16, out_channels)
 
+        # Add unused parameter:
+        self.param = torch.nn.Parameter(torch.empty(1))
+
     def forward(self, x, edge_index, batch=None, edge_label_index=None):
         x = self.conv1(x, edge_index).relu()
         x = self.conv2(x, edge_index)