refine documentation.

dmlc · Jul 18, 2024 · e5a9bc4 · e5a9bc4
1 parent 50285e2
commit e5a9bc4
Showing 1 changed file with 3 additions and 2 deletions.
diff --git a/python/dgl/graphbolt/impl/cpu_cached_feature.py b/python/dgl/graphbolt/impl/cpu_cached_feature.py
@@ -25,11 +25,12 @@ class CPUCachedFeature(Feature):
         The fallback feature.
     max_cache_size_in_bytes : int
         The capacity of the cache in bytes.
-    policy:
+    policy : str
         The cache eviction policy algorithm name. See gb.impl.CPUFeatureCache
         for the list of available policies.
-    pin_memory:
+    pin_memory : bool
         Whether the cache storage should be allocated on system pinned memory.
+        Default is False.
     """
 
     def __init__(