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)