DGL deepwalk weighted graph

Hi,
Can Dgl deepwalk work with weighted graph?

DeepWalk calls random_walk() which supports prob argument. Could you implement one on your own by referring to existing code?