update import

neuralmagic · Aug 1, 2024 · 26b6f80 · 26b6f80
1 parent 179fd90
commit 26b6f80
Showing 1 changed file with 1 addition and 1 deletion.
diff --git a/tests/sparseml/transformers/obcq/test_mask_structure_preservation.py b/tests/sparseml/transformers/obcq/test_mask_structure_preservation.py
@@ -19,7 +19,7 @@
 import pytest
 
 import sparseml
-from compressed_tensors.compressors.utils import tensor_follows_mask_structure
+from compressed_tensors.utils import tensor_follows_mask_structure
 from parameterized import parameterized_class
 from tests.testing_utils import parse_params, requires_torch