fix

Signed-off-by: Pawel Gadzinski <pgadzinski@nvidia.com>

Author: pggPL

transformer_engine/debug/features/disable_fp8_gemm.py

@@ -45,5 +45,3 @@ class DisableFP8GEMM(DisableQuantizationGEMM):
                     enabled: True
                     gemms: [dgrad, wgrad]
     """
-
-    pass  # Inherits all functionality from DisableQuantizationGEMM

transformer_engine/debug/features/disable_fp8_layer.py

@@ -34,5 +34,3 @@ class DisableFP8Layer(DisableQuantizationLayer):
                 DisableFP8Layer:  # Deprecated: use DisableQuantizationLayer
                     enabled: True
     """
-
-    pass  # Inherits all functionality from DisableQuantizationLayer

transformer_engine/debug/features/log_fp8_tensor_stats.py

@@ -9,12 +9,13 @@
 
 import torch
 import nvdlfw_inspect.api as debug_api
-
+import transformer_engine_torch as tex
 
 from nvdlfw_inspect.debug_features.log_tensor_stats import LogTensorStats as BaseLogTensorStats
 from nvdlfw_inspect.registry import Registry, api_method
 
 from transformer_engine.debug.features.utils.stats_buffer import STATS_BUFFERS
+from transformer_engine.debug.features.utils import get_reduction_params, next_enabled_iter
 from transformer_engine.pytorch.tensor import Quantizer, QuantizedTensor
 from transformer_engine.pytorch.tensor.float8_tensor import (
     Float8Quantizer,
@@ -23,8 +24,6 @@
 from transformer_engine.pytorch.tensor.mxfp8_tensor import MXFP8Quantizer
 from transformer_engine.pytorch.tensor.float8_blockwise_tensor import Float8BlockQuantizer
 
-import transformer_engine_torch as tex
-
 try:
     from transformer_engine.pytorch.tensor.nvfp4_tensor import NVFP4Quantizer
 
@@ -33,8 +32,6 @@
     _nvfp4_available = False
     NVFP4Quantizer = None
 
-from transformer_engine.debug.features.utils import get_reduction_params, next_enabled_iter
-
 
 ALL_RECIPE_NAMES = ["fp8_delayed_scaling", "fp8_current_scaling", "mxfp8", "fp8_block_scaling"]
 

transformer_engine/debug/features/log_nvfp4_tensor_stats.py

@@ -103,7 +103,7 @@ def update_aux_dict(
         self,
         aux_dict: Dict,
         quantized_tensor: QuantizedTensor,
-        quantizer: Quantizer,
+        quantizer: Quantizer,  # pylint: disable=unused-argument
         original_tensor: torch.Tensor,
     ):
         """