Unverified Commit 09e48552 authored by Muhammed Fatih BALIN's avatar Muhammed Fatih BALIN Committed by GitHub
Browse files

[GraphBolt] `to_pyg_data` should return `int64` `edge_index`. (#7178)

parent d8d87243
...@@ -516,7 +516,7 @@ class MiniBatch: ...@@ -516,7 +516,7 @@ class MiniBatch:
row_nodes = torch.cat(row_nodes) row_nodes = torch.cat(row_nodes)
edge_index = torch.unique( edge_index = torch.unique(
torch.stack((row_nodes, col_nodes)), dim=1 torch.stack((row_nodes, col_nodes)), dim=1
) ).long()
if self.node_features is None: if self.node_features is None:
node_features = None node_features = None
......
Markdown is supported
0% or .
You are about to add 0 people to the discussion. Proceed with caution.
Finish editing this message first!
Please register or to comment