fix

dmlc · Feb 19, 2024 · c7b76c2 · c7b76c2
1 parent 13cbad3
commit c7b76c2
Showing 1 changed file with 1 addition and 1 deletion.
diff --git 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: