typo's

neuralmagic · Jan 8, 2025 · e8f26d4 · e8f26d4
1 parent 5f5b5c3
commit e8f26d4
Show file tree

Hide file tree

Showing 2 changed files with 2 additions and 3 deletions.
diff --git a/src/compressed_tensors/compressors/sparse_compressors/base.py b/src/compressed_tensors/compressors/sparse_compressors/base.py
@@ -78,8 +78,7 @@ def compress(
             f"Compressing model with {len(model_state)} parameterized layers..."
         )
         for name, value in tqdm(model_state.items(), desc="Compressing model"):
-            ignored = not self.should_compress(name, compression_targets)
-            if ignored:
+            if not self.should_compress(name, compression_targets):
                 compressed_dict[name] = value
                 continue
             prefix = name

diff --git a/tests/testing_utils.py b/tests/testing_utils.py
@@ -130,7 +130,7 @@ def induce_sparsity(tensor, sparsity_ratio) -> "torch.Tensor":
 
 def is_gpu_available():
     """
-    Check for GPU and warn if not found
+    :return: True if a GPU is available, False otherwise
     """
     try:
         import torch  # noqa: F401