From 8e03d18dbebb60bd6c6a2d7419eeacb4f6db3fed Mon Sep 17 00:00:00 2001 From: SwiftieH Date: Wed, 29 Dec 2021 13:28:09 +0800 Subject: [PATCH] fix bug in pyg gin base.py --- test/performance/graph_classification/pyg/base.py | 2 +- 1 file changed, 1 insertion(+), 1 deletion(-) diff --git a/test/performance/graph_classification/pyg/base.py b/test/performance/graph_classification/pyg/base.py index 31c2f3c..ddd1453 100644 --- a/test/performance/graph_classification/pyg/base.py +++ b/test/performance/graph_classification/pyg/base.py @@ -10,7 +10,7 @@ import torch.nn.functional as F from torch.nn import Sequential, Linear, ReLU import torch_geometric from torch_geometric.datasets import TUDataset -if int(torch_geometric.__version__.split(",")[0]) >= 2: +if int(torch_geometric.__version__.split(".")[0]) >= 2: from torch_geometric.loader import DataLoader else: from torch_geometric.data import DataLoader