Skip to content

Commit

Permalink
Fix the bug.
Browse files Browse the repository at this point in the history
  • Loading branch information
mfbalin committed Mar 12, 2024
1 parent c48fd33 commit 0f820b8
Showing 1 changed file with 1 addition and 1 deletion.
2 changes: 1 addition & 1 deletion python/dgl/graphbolt/impl/neighbor_sampler.py
Original file line number Diff line number Diff line change
Expand Up @@ -361,7 +361,7 @@ def sampling_stages(
sampler, fanout, replace, prob_name
)
datapipe = datapipe.compact_per_layer(deduplicate)
if not layer_dependency:
if is_labor and not layer_dependency:
datapipe = datapipe.transform(self._increment_seed)
if is_labor:
datapipe = datapipe.transform(self._delattr_dependency)
Expand Down

0 comments on commit 0f820b8

Please sign in to comment.