Update dataloader.py (#3322)

75d793a1 · Quan (Andy) Gan · GitHub · f5183820 · 75d793a1
Unverified Commit 75d793a1 authored Sep 06, 2021 by Quan (Andy) Gan Committed by GitHub Sep 06, 2021
Show whitespace changes
Inline Side-by-side

Showing with 2 additions and 2 deletions

python/dgl/dataloading/dataloader.py python/dgl/dataloading/dataloader.py +2 -2

No files found.
--- a/python/dgl/dataloading/dataloader.py
+++ b/python/dgl/dataloading/dataloader.py
@@ -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,