From 26b6f80173103e097d8c8f23c613a76dcef94dae Mon Sep 17 00:00:00 2001 From: Sara Adkins Date: Thu, 1 Aug 2024 13:24:52 +0000 Subject: [PATCH] update import --- .../transformers/obcq/test_mask_structure_preservation.py | 2 +- 1 file changed, 1 insertion(+), 1 deletion(-) diff --git a/tests/sparseml/transformers/obcq/test_mask_structure_preservation.py b/tests/sparseml/transformers/obcq/test_mask_structure_preservation.py index eca6f5d2379..358213cf219 100644 --- 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