Skip to content

Commit 241760a

Browse files
[Graphbolt] Fix link_prediction example (#6397)
Co-authored-by: Hongzhi (Steve), Chen <[email protected]>
1 parent cc079e1 commit 241760a

File tree

2 files changed

+3
-2
lines changed

2 files changed

+3
-2
lines changed

CONTRIBUTORS.md

+1
Original file line numberDiff line numberDiff line change
@@ -36,6 +36,7 @@ Contributors
3636
* [Michal Szarmach](https://github.com/mszarma) from Intel
3737
* [Izabela Mazur](https://github.com/IzabelaMazur) from Intel
3838
* [Sanchit Misra](https://github.com/sanchit-misra) from Intel
39+
* [Andrzej Kotlowski](https://github.com/anko-intel) from Intel
3940
* [Sheng Zha](https://github.com/szha) from AWS
4041
* [Yifei Ma](https://github.com/yifeim) from AWS
4142
* [Yizhi Liu](https://github.com/yzhliu) from AWS

examples/sampling/graphbolt/link_prediction.py

+2-2
Original file line numberDiff line numberDiff line change
@@ -169,15 +169,15 @@ def create_dataloader(args, graph, features, itemset, is_train=True):
169169

170170
############################################################################
171171
# [Step-4]:
172-
# gb.to_dgl()
172+
# datapipe.to_dgl()
173173
# [Input]:
174174
# 'datapipe': The previous datapipe object.
175175
# [Output]:
176176
# A DGLMiniBatch used for computing.
177177
# [Role]:
178178
# Convert a mini-batch to dgl-minibatch.
179179
############################################################################
180-
datapipe = gb.to_dgl()
180+
datapipe = datapipe.to_dgl()
181181

182182
############################################################################
183183
# [Input]:

0 commit comments

Comments
 (0)