[Doc] fix examples (dmlc#3747)

sanchit-misra · Feb 17, 2022 · 80fb4db · 80fb4db
1 parent 9e358df
commit 80fb4db
Show file tree

Hide file tree

Showing 2 changed files with 12 additions and 12 deletions.
diff --git a/python/dgl/nn/pytorch/conv/appnpconv.py b/python/dgl/nn/pytorch/conv/appnpconv.py
@@ -45,17 +45,17 @@ class APPNPConv(nn.Module):
     >>> feat = th.ones(6, 10)
     >>> conv = APPNPConv(k=3, alpha=0.5)
     >>> res = conv(g, feat)
-    >>> res
-    tensor([[1.0000, 1.0000, 1.0000, 1.0000, 1.0000, 1.0000, 1.0000, 1.0000, 1.0000,
-            1.0000],
-            [1.0000, 1.0000, 1.0000, 1.0000, 1.0000, 1.0000, 1.0000, 1.0000, 1.0000,
-            1.0000],
-            [1.0000, 1.0000, 1.0000, 1.0000, 1.0000, 1.0000, 1.0000, 1.0000, 1.0000,
-            1.0000],
-            [1.0303, 1.0303, 1.0303, 1.0303, 1.0303, 1.0303, 1.0303, 1.0303, 1.0303,
-            1.0303],
-            [0.8643, 0.8643, 0.8643, 0.8643, 0.8643, 0.8643, 0.8643, 0.8643, 0.8643,
-            0.8643],
+    >>> print(res)
+    tensor([[0.8536, 0.8536, 0.8536, 0.8536, 0.8536, 0.8536, 0.8536, 0.8536, 0.8536,
+            0.8536],
+            [0.9268, 0.9268, 0.9268, 0.9268, 0.9268, 0.9268, 0.9268, 0.9268, 0.9268,
+            0.9268],
+            [0.9634, 0.9634, 0.9634, 0.9634, 0.9634, 0.9634, 0.9634, 0.9634, 0.9634,
+            0.9634],
+            [0.9268, 0.9268, 0.9268, 0.9268, 0.9268, 0.9268, 0.9268, 0.9268, 0.9268,
+            0.9268],
+            [0.9634, 0.9634, 0.9634, 0.9634, 0.9634, 0.9634, 0.9634, 0.9634, 0.9634,
+            0.9634],
             [0.5000, 0.5000, 0.5000, 0.5000, 0.5000, 0.5000, 0.5000, 0.5000, 0.5000,
             0.5000]])
     """

diff --git a/python/dgl/nn/pytorch/conv/sgconv.py b/python/dgl/nn/pytorch/conv/sgconv.py
@@ -75,7 +75,7 @@ class SGConv(nn.Module):
     >>> g = dgl.graph(([0,1,2,3,2,5], [1,2,3,4,0,3]))
     >>> g = dgl.add_self_loop(g)
     >>> feat = th.ones(6, 10)
-    >>> conv = SGConv(10, 2, k=2, cached=True)
+    >>> conv = SGConv(10, 2, k=2)
     >>> res = conv(g, feat)
     >>> res
     tensor([[-1.9441, -0.9343],