fix (dmlc#3076)

lizeyan · Jun 30, 2021 · a97aaa4 · a97aaa4
1 parent 7823ae2
commit a97aaa4
Showing 1 changed file with 2 additions and 0 deletions.
diff --git a/python/dgl/sampling/node2vec_randomwalk.py b/python/dgl/sampling/node2vec_randomwalk.py
@@ -20,10 +20,12 @@ def node2vec_random_walk(g, nodes, p, q, walk_length, prob=None, return_eids=Fal
 
     Note that if a random walk stops in advance, DGL pads the trace with -1 to have the same
     length.
+
     Parameters
     ----------
     g : DGLGraph
         The graph.  Must be on CPU.
+
         Note that node2vec only support homogeneous graph.
     nodes : Tensor
         Node ID tensor from which the random walk traces starts.