Unverified Commit 63bc44da authored by Hongzhi (Steve), Chen's avatar Hongzhi (Steve), Chen Committed by GitHub
Browse files

[Graphbolt] Pass down the original eid. (#6444)


Co-authored-by: default avatarUbuntu <ubuntu@ip-172-31-28-63.ap-northeast-1.compute.internal>
parent f08d2a8a
...@@ -124,6 +124,7 @@ class NeighborSampler(SubgraphSampler): ...@@ -124,6 +124,7 @@ class NeighborSampler(SubgraphSampler):
node_pairs=compacted_node_pairs, node_pairs=compacted_node_pairs,
original_column_node_ids=original_column_node_ids, original_column_node_ids=original_column_node_ids,
original_row_node_ids=seeds, original_row_node_ids=seeds,
original_edge_ids=subgraph.original_edge_ids,
) )
subgraphs.insert(0, subgraph) subgraphs.insert(0, subgraph)
return seeds, subgraphs return seeds, subgraphs
......
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