Skip to content

GitLab

Unverified Commit 53153b42 authored by peizhou001's avatar peizhou001 Committed by GitHub
Browse files

[Graphbolt] Fix sample bugs (#6311)

parent 337b5ea7
Hide whitespace changes
Inline Side-by-side
Showing with 3 additions and 3 deletions
python/dgl/graphbolt/impl/neighbor_sampler.py
View file @ 53153b42
...@@ -102,14 +102,14 @@ class NeighborSampler(SubgraphSampler): ...@@ -102,14 +102,14 @@ class NeighborSampler(SubgraphSampler):
self.replace, self.replace,
self.prob_name, self.prob_name,
) )
reverse_row_node_ids = seeds reverse_column_node_ids = seeds
seeds, compacted_node_pairs = unique_and_compact_node_pairs( seeds, compacted_node_pairs = unique_and_compact_node_pairs(
subgraph.node_pairs, seeds subgraph.node_pairs, seeds
) )
subgraph = SampledSubgraphImpl( subgraph = SampledSubgraphImpl(
node_pairs=compacted_node_pairs, node_pairs=compacted_node_pairs,
reverse_column_node_ids=seeds, reverse_column_node_ids=reverse_column_node_ids,
reverse_row_node_ids=reverse_row_node_ids, reverse_row_node_ids=seeds,
) )
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