Merge branch 'gb_non_blocking_copy_to' into gb_dataloader_non_blockin…

…g_copy_to
dmlc · Jul 27, 2024 · 0df5a21 · 0df5a21
2 parents f268362 + 13e9a68
commit 0df5a21
Showing 1 changed file with 2 additions and 2 deletions.
diff --git a/python/dgl/graphbolt/base.py b/python/dgl/graphbolt/base.py
@@ -491,9 +491,9 @@ def __init__(self, indptr: torch.Tensor, indices: torch.Tensor):
     def __repr__(self) -> str:
         return _csc_format_base_str(self)
 
-    def to(
+    def to(  # pylint: disable=invalid-name
         self, device: torch.device, non_blocking=False
-    ) -> None:  # pylint: disable=invalid-name
+    ) -> None:
         """Copy `CSCFormatBase` to the specified device using reflection."""
 
         for attr in dir(self):