Skip to content

GitLab

"...api/git@developer.sourcefind.cn:renzhc/diffusers_dcu.git" did not exist on "10b4e354b6d1f8e85279924e8f468bd572974249"
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):
self.replace,
self.prob_name,
)
reverse_row_node_ids = seeds
reverse_column_node_ids = seeds
seeds, compacted_node_pairs = unique_and_compact_node_pairs(
subgraph.node_pairs, seeds
)
subgraph = SampledSubgraphImpl(
node_pairs=compacted_node_pairs,
reverse_column_node_ids=seeds,
reverse_row_node_ids=reverse_row_node_ids,
reverse_column_node_ids=reverse_column_node_ids,
reverse_row_node_ids=seeds,
)
subgraphs.insert(0, subgraph)
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