typo

pyg-team · Dec 19, 2022 · a986d59 · a986d59
1 parent d02c180
commit a986d59
Showing 1 changed file with 1 addition and 1 deletion.
diff --git a/test/explain/metric/test_fidelity.py b/test/explain/metric/test_fidelity.py
@@ -17,7 +17,7 @@ def forward(self, x, edge_index):
 
 @pytest.mark.parametrize('explanation_type', ['model', 'phenomenon'])
 def test_fidelity(explanation_type):
-    x = torch.randn(8, 1)
+    x = torch.randn(8, 4)
     edge_index = torch.tensor([
         [0, 1, 1, 2, 2, 3, 3, 4, 4, 5, 5, 6, 6, 7],
         [1, 0, 2, 1, 3, 2, 4, 3, 5, 4, 6, 5, 7, 6],