update rst

rapidsai · May 14, 2024 · 6dc5e09 · 6dc5e09
1 parent aed0555
commit 6dc5e09
Showing 1 changed file with 12 additions and 2 deletions.
diff --git a/docs/cugraph/source/api_docs/cugraph-pyg/cugraph_pyg.rst b/docs/cugraph/source/api_docs/cugraph-pyg/cugraph_pyg.rst
@@ -9,5 +9,15 @@ cugraph-pyg
 .. autosummary::
    :toctree: ../api/cugraph-pyg/
 
-..   cugraph_pyg.data.cugraph_store.EXPERIMENTAL__CuGraphStore
-..   cugraph_pyg.sampler.cugraph_sampler.EXPERIMENTAL__CuGraphSampler
+..   cugraph_pyg.data.dask_graph_store.DaskGraphStore
+..   cugraph_pyg.data.graph_store.GraphStore
+..   cugraph_pyg.data.feature_store.TensorDictFeatureStore
+..   cugraph_pyg.loader.dask_node_loader.DaskNeighborLoader
+..   cugraph_pyg.loader.dask_node_loader.BulkSampleLoader
+..   cugraph_pyg.loader.node_loader.NodeLoader
+..   cugraph_pyg.loader.neighbor_loader.NeighborLoader
+..   cugraph_pyg.sampler.sampler.BaseSampler
+..   cugraph_pyg.sampler.sampler.SampleReader
+..   cugraph_pyg.sampler.sampler.HomogeneousSampleReader
+..   cugraph_pyg.sampler.sampler.SampleIterator
+