Skip to content
GitLab
Menu
Projects
Groups
Snippets
Loading...
Help
Help
Support
Community forum
Keyboard shortcuts
?
Submit feedback
Contribute to GitLab
Sign in / Register
Toggle navigation
Menu
Open sidebar
OpenDAS
dgl
Commits
5d4f6bca
"examples/git@developer.sourcefind.cn:OpenDAS/dgl.git" did not exist on "56b6a4ebd8bc8e2d2d7b21174f833330ca748ab0"
Unverified
Commit
5d4f6bca
authored
Oct 14, 2021
by
Quan (Andy) Gan
Committed by
GitHub
Oct 14, 2021
Browse files
[Fix] Fix edge ID exclusion not working in EdgeDataLoader (#3412)
parent
8798872f
Changes
1
Hide whitespace changes
Inline
Side-by-side
Showing
1 changed file
with
2 additions
and
1 deletion
+2
-1
python/dgl/dataloading/dataloader.py
python/dgl/dataloading/dataloader.py
+2
-1
No files found.
python/dgl/dataloading/dataloader.py
View file @
5d4f6bca
...
...
@@ -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
:
...
...
Write
Preview
Markdown
is supported
0%
Try again
or
attach a new file
.
Attach a file
Cancel
You are about to add
0
people
to the discussion. Proceed with caution.
Finish editing this message first!
Cancel
Please
register
or
sign in
to comment