[Doc] fix display issue of node/edge_split (dmlc#3858)

Co-authored-by: Jinjing Zhou <[email protected]>
skorani · Mar 25, 2022 · 16409ff · 16409ff
1 parent a0e8cf0
commit 16409ff
Showing 1 changed file with 2 additions and 0 deletions.
diff --git a/docs/source/api/python/dgl.distributed.rst b/docs/source/api/python/dgl.distributed.rst
@@ -45,6 +45,8 @@ Distributed embedding optimizer
 Distributed workload split
 --------------------------
 
+.. currentmodule:: dgl.distributed.dist_graph
+
 .. autosummary::
     :toctree: ../../generated/