From 5b81af41b00b7982cfe9cfe7e07102b8d180da84 Mon Sep 17 00:00:00 2001 From: rusty1s Date: Mon, 28 Aug 2023 12:40:15 +0000 Subject: [PATCH] update --- test/loader/test_cache.py | 2 +- test/nn/models/test_basic_gnn.py | 1 - 2 files changed, 1 insertion(+), 2 deletions(-) diff --git a/test/loader/test_cache.py b/test/loader/test_cache.py index 743c87aa2e3d..7c15b2e94a94 100644 --- 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 index 4ce3d727c6cb..25e175bb34e1 100644 --- 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,