"src/git@developer.sourcefind.cn:renzhc/diffusers_dcu.git" did not exist on "6683f97959dabb35feacba1d41db3a1c6296d2f6"
Unverified Commit b81bb914 authored by Quan (Andy) Gan's avatar Quan (Andy) Gan Committed by GitHub
Browse files

[Bug] Fix edge exclusion still not working for full neighbor sampling (#3424)

parent 4f5c3aa2
......@@ -386,7 +386,7 @@ class BlockSampler(object):
# Removing edges from the frontier for link prediction training falls
# into the category of frontier postprocessing
if not self.exclude_edges_in_frontier:
if not self.exclude_edges_in_frontier(g):
eid_excluder = _create_eid_excluder(exclude_eids, self.output_device)
if eid_excluder is not None:
frontier = eid_excluder(frontier)
......
......@@ -165,3 +165,7 @@ class MultiLayerFullNeighborSampler(MultiLayerNeighborSampler):
"""
def __init__(self, n_layers, return_eids=False):
super().__init__([None] * n_layers, return_eids=return_eids)
@classmethod
def exclude_edges_in_frontier(cls, g):
return False
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