"examples/git@developer.sourcefind.cn:OpenDAS/dgl.git" did not exist on "56b6a4ebd8bc8e2d2d7b21174f833330ca748ab0"
Unverified Commit 5d4f6bca authored by Quan (Andy) Gan's avatar Quan (Andy) Gan Committed by GitHub
Browse files

[Fix] Fix edge ID exclusion not working in EdgeDataLoader (#3412)

parent 8798872f
......@@ -108,6 +108,7 @@ class _EidExcluder():
frontier, v, etype=k, store_ids=True)
new_eids[k] = F.gather_row(parent_eids[k], frontier.edges[k].data[EID])
frontier.edata[EID] = new_eids
return frontier
def _create_eid_excluder(exclude_eids, device):
......@@ -388,7 +389,7 @@ class BlockSampler(object):
if not self.exclude_edges_in_frontier:
eid_excluder = _create_eid_excluder(exclude_eids, self.output_device)
if eid_excluder is not None:
eid_excluder(frontier)
frontier = eid_excluder(frontier)
block = transform.to_block(frontier, seed_nodes_out)
if self.return_eids:
......
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