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