update rst

docs/cugraph/source/api_docs/cugraph-pyg/cugraph_pyg.rst

@@ -6,18 +6,37 @@ cugraph-pyg
 
 .. currentmodule:: cugraph_pyg
 
+Graph Storage
+-------------
 .. autosummary::
    :toctree: ../api/cugraph-pyg/
 
 ..   cugraph_pyg.data.dask_graph_store.DaskGraphStore
 ..   cugraph_pyg.data.graph_store.GraphStore
+
+Feature Storage
+---------------
+.. autosummary::
+   :toctree: ../api/cugraph-pyg/
+
 ..   cugraph_pyg.data.feature_store.TensorDictFeatureStore
+
+Data Loaders
+------------
+.. autosummary::
+   :toctree: ../api/cugraph-pyg/
+
 ..   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
+
+Samplers
+--------
+.. autosummary::
+   :toctree: ../api/cugraph-pyg/
+
 ..   cugraph_pyg.sampler.sampler.BaseSampler
 ..   cugraph_pyg.sampler.sampler.SampleReader
 ..   cugraph_pyg.sampler.sampler.HomogeneousSampleReader
 ..   cugraph_pyg.sampler.sampler.SampleIterator
-