Update hetero_link_pred.py
example to support MPS
#7784
Merged
hetero_link_pred.py
example to support MPS
#7784