fixed test

pyg-team · Mar 27, 2019 · f07ef32 · f07ef32
1 parent a8aece6
commit f07ef32
Show file tree

Hide file tree

Showing 2 changed files with 2 additions and 2 deletions.
diff --git a/.gitignore b/.gitignore
@@ -7,6 +7,7 @@ alpha/
 .eggs/
 *.egg-info/
 .coverage
+.coverage.*
 
 !torch_geometric/data/
 !test/data/

diff --git a/test/nn/models/test_autoencoder.py b/test/nn/models/test_autoencoder.py
@@ -32,8 +32,7 @@ def test_gae():
     assert data.train_neg_adj_mask.sum().item() == (11**2 - 11) / 2 - 4 - 6 - 5
 
     z = torch.randn(11, 16)
-    loss = model.recon_loss(z, data.train_pos_edge_index,
-                            data.train_neg_adj_mask)
+    loss = model.recon_loss(z, data.train_pos_edge_index)
     assert loss.item() > 0
 
     auc, ap = model.test(z, data.val_pos_edge_index, data.val_neg_edge_index)