[Doc] Fixing KNNGraph docs (#803)

828e269c · Quan (Andy) Gan · GitHub · e17add56 · 828e269c · 828e269c
Unverified Commit 828e269c authored Aug 28, 2019 by Quan (Andy) Gan Committed by GitHub Aug 28, 2019
Show whitespace changes
Inline Side-by-side

Showing with 4 additions and 4 deletions

docs/source/api/python/nn.pytorch.rst docs/source/api/python/nn.pytorch.rst +2 -2

docs/source/api/python/transform.rst docs/source/api/python/transform.rst +2 -2

No files found.
--- a/docs/source/api/python/nn.pytorch.rst
+++ b/docs/source/api/python/nn.pytorch.rst
@@ -121,10 +121,10 @@ dgl.nn.pytorch.softmax
 dgl.nn.pytorch.factory
 ----------------------

-.. automodule:: dgl.nn.pytorch.NearestNeighborGraph
+.. automodule:: dgl.nn.pytorch.KNNGraph
    :members:
    :show-inheritance:

-.. automodule:: dgl.nn.pytorch.SegmentedNearestNeighborGraph
+.. automodule:: dgl.nn.pytorch.SegmentedKNNGraph
    :members:
    :show-inheritance:
--- a/docs/source/api/python/transform.rst
+++ b/docs/source/api/python/transform.rst
@@ -15,5 +15,5 @@ Transform -- Graph Transformation
    khop_adj
    khop_graph
    laplacian_lambda_max
-    nearest_neighbor_graph
-    segmented_nearest_neighbor_graph
+    knn_graph
+    segmented_knn_graph