diff --git a/docs/source/api/python/sampler.rst b/docs/source/api/python/sampler.rst
index 839db52d5a..d73d4afe64 100644
--- a/docs/source/api/python/sampler.rst
+++ b/docs/source/api/python/sampler.rst
@@ -5,6 +5,7 @@ Graph samplers
 
 .. autofunction:: dgl.contrib.sampling.sampler.NeighborSampler
 .. autofunction:: dgl.contrib.sampling.sampler.LayerSampler
+.. autofunction:: dgl.contrib.sampling.sampler.EdgeSampler
 
 Distributed sampler
 ------------------------