fix style

Signed-off-by: shanjiaz <zsjwpianpian@gmail.com>

Author: shanjiaz

src/compressed_tensors/compressors/sparse_compressors/base.py

@@ -25,7 +25,6 @@
 from torch import Tensor
 from tqdm import tqdm
 
-from torch.nn import Module
 
 
 __all__ = ["BaseSparseCompressor"]
@@ -96,7 +95,6 @@ def compress(
                 prefix = prefix[: -(len(".weight"))]
 
             compression_data = self.compress_weight(prefix, value)
-
             for key in compression_data.keys():
                 if key in compressed_dict:
                     _LOGGER.warn(

src/compressed_tensors/compressors/sparse_compressors/sparse_24_bitmask.py

@@ -56,7 +56,6 @@ def compress_weight(self, name, value):
         bitmask_tensor = Sparse24BitMaskTensor.from_dense(
             value, self.config.sparsity_structure
         )
-        # Normal flow: return compression dict
         return bitmask_tensor.dict(
             name_prefix=name,
             device="meta" if value.device.type == "meta" else "cpu",

src/compressed_tensors/compressors/sparse_compressors/sparse_bitmask.py

@@ -46,7 +46,7 @@ def compression_param_names(self) -> Tuple[str]:
         """
         return ("shape", "compressed", "bitmask", "row_offsets")
 
-    def compress_weight(self, name, value, **kwargs):
+    def compress_weight(self, name, value):
         bitmask_tensor = BitmaskTensor.from_dense(value)
         bitmask_dict = bitmask_tensor.dict(name_prefix=name, device="cpu")
         return bitmask_dict