Remove remaining llmcompressor.transformers.sparsification reference (#1845)

fynnsu · web-flow · commit eab8bf7a544e · 2025-09-18T21:29:56.000-04:00
SUMMARY: `llmcompressor.transformers.sparsification` was removed in #1826 but this reference was missed. Signed-off-by: Fynn Schmitt-Ulms <fschmitt@redhat.com>
diff --git a/tests/llmcompressor/transformers/obcq/test_obcq_completion.py b/tests/llmcompressor/transformers/obcq/test_obcq_completion.py
@@ -51,7 +51,7 @@ def _test_oneshot_completion(self, model_name: str = None):
         from llmcompressor import oneshot
         from llmcompressor.pytorch.model_load.helpers import get_session_model
         from llmcompressor.pytorch.utils import tensors_to_device
-        from llmcompressor.transformers.sparsification.compressed_tensors_utils import (
+        from llmcompressor.transformers.compression.compressed_tensors_utils import (
             get_model_compressor,  # avoid circular import
         )
 

Original file line number	Diff line number	Diff line change
`@@ -51,7 +51,7 @@ def _test_oneshot_completion(self, model_name: str = None):`
`51`	`51`	`from llmcompressor import oneshot`
`52`	`52`	`from llmcompressor.pytorch.model_load.helpers import get_session_model`
`53`	`53`	`from llmcompressor.pytorch.utils import tensors_to_device`
`54`		`- from llmcompressor.transformers.sparsification.compressed_tensors_utils import (`
	`54`	`+ from llmcompressor.transformers.compression.compressed_tensors_utils import (`
`55`	`55`	`get_model_compressor, # avoid circular import`
`56`	`56`	`)`
`57`	`57`