Update a docstring

neuralmagic · Dec 17, 2024 · 9eeede7 · 9eeede7
1 parent 6936121
commit 9eeede7
Showing 1 changed file with 2 additions and 1 deletion.
diff --git a/src/compressed_tensors/compressors/sparse_compressors/base.py b/src/compressed_tensors/compressors/sparse_compressors/base.py
@@ -129,7 +129,8 @@ def decompress(
     @staticmethod
     def should_compress(name: str, expanded_targets: Optional[Set[str]] = None) -> bool:
         """
-        Check if a parameter should be compressed
+        Check if a parameter should be compressed.
+        Currently, this only returns True for weight parameters.
 
         :param name: name of the parameter
         :param expanded_targets: set of layer prefixes to compress