[Doc] Fix the example code of negative sampler (dmlc#3501)

Co-authored-by: Jinjing Zhou <[email protected]>
skorani · Nov 16, 2021 · 65fdfad · 65fdfad
1 parent b8ce0f4
commit 65fdfad
Showing 1 changed file with 1 addition and 1 deletion.
diff --git a/python/dgl/dataloading/negative_sampler.py b/python/dgl/dataloading/negative_sampler.py
@@ -49,7 +49,7 @@ class Uniform(_BaseNegativeSampler):
     --------
     >>> g = dgl.graph(([0, 1, 2], [1, 2, 3]))
     >>> neg_sampler = dgl.dataloading.negative_sampler.Uniform(2)
-    >>> neg_sampler(g, [0, 1])
+    >>> neg_sampler(g, torch.tensor([0, 1]))
     (tensor([0, 0, 1, 1]), tensor([1, 0, 2, 3]))
     """
     def __init__(self, k):