Merge branch 'master' into datapipe

dmlc · Aug 2, 2024 · 865f073 · 865f073
2 parents 96270e7 + f1c28d3
commit 865f073
Show file tree

Hide file tree

Showing 2 changed files with 5 additions and 3 deletions.
diff --git a/docs/source/api/python/dgl.graphbolt.rst b/docs/source/api/python/dgl.graphbolt.rst
@@ -5,7 +5,7 @@
 
 .. currentmodule:: dgl.graphbolt
 
-**dgl.graphbolt** is a dataloading framework for GNN that provides well-defined
+**dgl.graphbolt** is a dataloading framework for GNNs that provides well-defined
 APIs for each stage of the data pipeline and multiple standard implementations.
 
 Dataset
@@ -55,6 +55,8 @@ collection of features.
     BasicFeatureStore
     TorchBasedFeature
     TorchBasedFeatureStore
+    DiskBasedFeature
+    CPUCachedFeature
     GPUCachedFeature
 
 

diff --git a/python/dgl/graphbolt/impl/gpu_cached_feature.py b/python/dgl/graphbolt/impl/gpu_cached_feature.py
@@ -89,7 +89,7 @@ def read(self, ids: torch.Tensor = None):
         return values
 
     def read_async(self, ids: torch.Tensor):
-        """Read the feature by index asynchronously.
+        r"""Read the feature by index asynchronously.
 
         Parameters
         ----------
@@ -104,7 +104,7 @@ def read_async(self, ids: torch.Tensor):
             can be accessed by calling `.wait()`. on the returned future object.
             It is undefined behavior to call `.wait()` more than once.
 
-        Example Usage
+        Examples
         --------
         >>> import dgl.graphbolt as gb
         >>> feature = gb.Feature(...)