Skip to content

GitLab

Unverified Commit 75d793a1 authored by Quan (Andy) Gan's avatar Quan (Andy) Gan Committed by GitHub
Browse files

Update dataloader.py (#3322)

parent f5183820
Hide whitespace changes
Inline Side-by-side
Showing with 2 additions and 2 deletions
python/dgl/dataloading/dataloader.py
View file @ 75d793a1
...@@ -739,7 +739,7 @@ class EdgeCollator(Collator): ...@@ -739,7 +739,7 @@ class EdgeCollator(Collator):
seed_nodes = pair_graph.ndata[NID] seed_nodes = pair_graph.ndata[NID]
exclude_eids = _find_exclude_eids( exclude_eids = _find_exclude_eids(
self.g, self.g_sampling,
self.exclude, self.exclude,
items, items,
reverse_eid_map=self.reverse_eids, reverse_eid_map=self.reverse_eids,
...@@ -782,7 +782,7 @@ class EdgeCollator(Collator): ...@@ -782,7 +782,7 @@ class EdgeCollator(Collator):
seed_nodes = pair_graph.ndata[NID] seed_nodes = pair_graph.ndata[NID]
exclude_eids = _find_exclude_eids( exclude_eids = _find_exclude_eids(
self.g, self.g_sampling,
self.exclude, self.exclude,
items, items,
reverse_eid_map=self.reverse_eids, reverse_eid_map=self.reverse_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