Skip to content

GitLab

"git@developer.sourcefind.cn:OpenDAS/mmcv.git" did not exist on "c2c9fced2bdd5f80b254723263beca0efb60166c"
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):
seed_nodes = pair_graph.ndata[NID]
exclude_eids = _find_exclude_eids(
self.g,
self.g_sampling,
self.exclude,
items,
reverse_eid_map=self.reverse_eids,
......@@ -782,7 +782,7 @@ class EdgeCollator(Collator):
seed_nodes = pair_graph.ndata[NID]
exclude_eids = _find_exclude_eids(
self.g,
self.g_sampling,
self.exclude,
items,
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