Fix an imort

vllm-project · Oct 4, 2024 · 7b6cb07 · 7b6cb07
1 parent 8b14532
commit 7b6cb07
Showing 1 changed file with 1 addition and 1 deletion.
diff --git a/tests/llmcompressor/transformers/finetune/test_oneshot_and_finetune.py b/tests/llmcompressor/transformers/finetune/test_oneshot_and_finetune.py
@@ -3,7 +3,7 @@
 import unittest
 
 import pytest
-from compressed_tensors.compressors.model_compressor import ModelCompressor
+from compressed_tensors.compressors import ModelCompressor
 from parameterized import parameterized_class
 from transformers import AutoConfig