fix test

pyg-team · Oct 25, 2022 · 69c8c6c · 69c8c6c
1 parent fb80ad1
commit 69c8c6c
Showing 1 changed file with 1 addition and 1 deletion.
diff --git a/test/loader/test_link_neighbor_loader.py b/test/loader/test_link_neighbor_loader.py
@@ -178,7 +178,7 @@ def test_link_neighbor_loader_edge_label():
 
     for batch in loader:
         assert batch.edge_label.dtype == torch.long
-        assert torch.all(batch.edge_label[:10] == 2)
+        assert torch.all(batch.edge_label[:10] == 1)
         assert torch.all(batch.edge_label[10:] == 0)