From c7b76c281b897012352c42013f40ef4c5eb2081b Mon Sep 17 00:00:00 2001 From: Ramon Zhou Date: Mon, 19 Feb 2024 08:40:33 +0000 Subject: [PATCH] fix --- python/dgl/graphbolt/minibatch.py | 2 +- 1 file changed, 1 insertion(+), 1 deletion(-) diff --git a/python/dgl/graphbolt/minibatch.py b/python/dgl/graphbolt/minibatch.py index aef6b31988a2..f75c72ab005c 100644 --- a/python/dgl/graphbolt/minibatch.py +++ b/python/dgl/graphbolt/minibatch.py @@ -500,7 +500,7 @@ def to_pyg_data(self): col_nodes = torch.cat(col_nodes) row_nodes = torch.cat(row_nodes) edge_index = torch.unique( - torch.stack((col_nodes, row_nodes)), dim=1 + torch.stack((row_nodes, col_nodes)), dim=1 ) if self.node_features is None: