Skip to content

Commit

Permalink
fix the example.
Browse files Browse the repository at this point in the history
  • Loading branch information
mfbalin committed Aug 29, 2024
1 parent 232139b commit a664059
Showing 1 changed file with 1 addition and 1 deletion.
2 changes: 1 addition & 1 deletion examples/graphbolt/link_prediction.py
Original file line number Diff line number Diff line change
Expand Up @@ -204,7 +204,7 @@ def create_dataloader(args, graph, features, itemset, is_train=True):
if is_train and args.exclude_edges:
datapipe = datapipe.exclude_seed_edges(
include_reverse_edges=True,
asynchronous=args.graph_device != "cpu",
asynchronous=args.storage_device != "cpu",
)

############################################################################
Expand Down

0 comments on commit a664059

Please sign in to comment.