fix lint error

Signed-off-by: Youngeun Kwon <[email protected]>
NVIDIA · Dec 5, 2024 · ff6d1d6 · ff6d1d6
1 parent 4e7694d
commit ff6d1d6
Showing 1 changed file with 4 additions and 1 deletion.
diff --git a/transformer_engine/pytorch/tensor/float8_tensor.py b/transformer_engine/pytorch/tensor/float8_tensor.py
@@ -443,7 +443,9 @@ def __new__(
 
         return self
 
-    def fsdp_pre_all_gather(self, mesh):  # pylint: disable=unused-argument
+    def fsdp_pre_all_gather(self, mesh):  
+        # pylint: disable=missing-function-docstring
+
         return (self._data,), (self,)
 
     def fsdp_post_all_gather(
@@ -454,6 +456,7 @@ def fsdp_post_all_gather(
         *,
         out: Optional[torch.Tensor] = None,
     ):
+        # pylint: disable=unused-argument
         (data,) = all_gather_outputs
         (sample,) = metadata
         if out is not None: