add type of last arg.

dmlc · Jul 18, 2024 · 50285e2 · 50285e2
1 parent 0145eae
commit 50285e2
Showing 1 changed file with 1 addition and 1 deletion.
diff --git a/python/dgl/graphbolt/impl/cpu_cached_feature.py b/python/dgl/graphbolt/impl/cpu_cached_feature.py
@@ -37,7 +37,7 @@ def __init__(
         fallback_feature: Feature,
         max_cache_size_in_bytes: int,
         policy: str = None,
-        pin_memory=False,
+        pin_memory: bool = False,
     ):
         super(CPUCachedFeature, self).__init__()
         assert isinstance(fallback_feature, Feature), (