From 9601a3c18d61f225c1e8f0b4571f2c41c755f4c0 Mon Sep 17 00:00:00 2001 From: mhz Date: Fri, 5 Jul 2024 10:49:43 +0200 Subject: [PATCH] comment edgelist --- graph_dit/datasets/dataset.py | 4 ++-- 1 file changed, 2 insertions(+), 2 deletions(-) diff --git a/graph_dit/datasets/dataset.py b/graph_dit/datasets/dataset.py index 0f5bb55..5004f2d 100644 --- a/graph_dit/datasets/dataset.py +++ b/graph_dit/datasets/dataset.py @@ -692,8 +692,8 @@ class Dataset(InMemoryDataset): if adj[start][end] == 1: edges_list.append((start, end)) edge_type.append(1) - edges_list.append((end, start)) - edge_type.append(1) + # edges_list.append((end, start)) + # edge_type.append(1) edge_index = torch.tensor(edges_list, dtype=torch.long).t() edge_type = torch.tensor(edge_type, dtype=torch.long)