update

pyg-team · Aug 28, 2023 · 5b81af4 · 5b81af4
1 parent 4ca3c6e
commit 5b81af4
Show file tree

Hide file tree

Showing 2 changed files with 1 addition and 2 deletions.
diff --git a/test/loader/test_cache.py b/test/loader/test_cache.py
@@ -72,7 +72,7 @@ def test_cached_loader_transform(device):
         assert batch.dim() == 2 and batch.size(0) == 2
         assert batch.device == device
 
-        cache.append(edge_index)
+        cache.append(batch)
 
     for i, batch in enumerate(cached_loader):
         assert torch.equal(batch, cache[i])
diff --git a/test/nn/models/test_basic_gnn.py b/test/nn/models/test_basic_gnn.py
@@ -13,7 +13,6 @@
 from torch_geometric.nn import SAGEConv
 from torch_geometric.nn.models import GAT, GCN, GIN, PNA, EdgeCNN, GraphSAGE
 from torch_geometric.profile import benchmark
-from torch_geometric.seed import seed_everything
 from torch_geometric.testing import (
     disableExtensions,
     onlyFullTest,