Update torch_geometric/datasets/heterophilous_graph_dataset.py

Co-authored-by: Jintang Li <cnljt@outlook.com>
pyg-team · Mar 29, 2023 · 21917db · 21917db
1 parent 7f5d45e
commit 21917db
Showing 1 changed file with 1 addition and 1 deletion.
diff --git a/torch_geometric/datasets/heterophilous_graph_dataset.py b/torch_geometric/datasets/heterophilous_graph_dataset.py
@@ -119,7 +119,7 @@ def process(self):
         data = Data(x=x, y=y, edge_index=edge_index, train_mask=train_mask,
                     val_mask=val_mask, test_mask=test_mask)
 
-        data = ToUndirected()(data)
+        data.edge_index = to_undirected(data.edge_index)
 
         if self.pre_transform is not None:
             data = self.pre_transform(data)