fix the lint and naming

dmlc · Aug 21, 2024 · 66bc1f6 · 66bc1f6
1 parent e26eccc
commit 66bc1f6
Show file tree

Hide file tree

Showing 2 changed files with 6 additions and 4 deletions.
diff --git a/python/dgl/graphbolt/impl/cpu_cached_feature.py b/python/dgl/graphbolt/impl/cpu_cached_feature.py
@@ -437,7 +437,7 @@ def update(self, value: torch.Tensor, ids: torch.Tensor = None):
             feat0 = value[:1]
             self._fallback_feature.update(value)
             cache_size = min(
-                bytes_to_number_of_items(self.max_cache_size_in_bytes, feat0),
+                bytes_to_number_of_items(self.cache_size_in_bytes, feat0),
                 value.shape[0],
             )
             self._feature = None  # Destroy the existing cache first.
@@ -452,7 +452,8 @@ def is_pinned(self):
         return self._feature.is_pinned()
 
     @property
-    def max_cache_size_in_bytes(self):
+    def cache_size_in_bytes(self):
+        """Return the size taken by the cache in bytes."""
         return self._feature.max_size_in_bytes
 
     @property

diff --git a/python/dgl/graphbolt/impl/gpu_cached_feature.py b/python/dgl/graphbolt/impl/gpu_cached_feature.py
@@ -237,7 +237,7 @@ def update(self, value: torch.Tensor, ids: torch.Tensor = None):
             feat0 = value[:1]
             self._fallback_feature.update(value)
             cache_size = min(
-                bytes_to_number_of_items(self.max_cache_size_in_bytes, feat0),
+                bytes_to_number_of_items(self.cache_size_in_bytes, feat0),
                 value.shape[0],
             )
             self._feature = None  # Destroy the existing cache first.
@@ -249,7 +249,8 @@ def update(self, value: torch.Tensor, ids: torch.Tensor = None):
             self._feature.replace(ids, value)
 
     @property
-    def max_cache_size_in_bytes(self):
+    def cache_size_in_bytes(self):
+        """Return the size taken by the cache in bytes."""
         return self._feature.max_size_in_bytes
 
     @property