[doc] fix module typo (dmlc#3734)

Co-authored-by: Mufei Li <[email protected]>
sanchit-misra · Feb 16, 2022 · 42f8c8f · 42f8c8f
1 parent 97a6d74
commit 42f8c8f
Showing 1 changed file with 1 addition and 1 deletion.
diff --git a/python/dgl/nn/pytorch/link/transr.py b/python/dgl/nn/pytorch/link/transr.py
@@ -50,7 +50,7 @@ class TransR(nn.Module):
     >>> num_rels = 3
     >>> feats = 4
 
-    >>> scorer = TransE(num_rels=num_rels, rfeats=2, nfeats=feats)
+    >>> scorer = TransR(num_rels=num_rels, rfeats=2, nfeats=feats)
     >>> g = dgl.rand_graph(num_nodes=num_nodes, num_edges=num_edges)
     >>> src, dst = g.edges()
     >>> h = th.randn(num_nodes, feats)