import ModelCompressor for type checking

Signed-off-by: Kyle Sayers <kylesayrs@gmail.com>
neuralmagic · Dec 2, 2024 · bed83b6 · bed83b6
1 parent 2dcbc9d
commit bed83b6
Showing 1 changed file with 5 additions and 1 deletion.
diff --git a/src/compressed_tensors/utils/helpers.py b/src/compressed_tensors/utils/helpers.py
@@ -12,12 +12,16 @@
 # See the License for the specific language governing permissions and
 # limitations under the License.
 
-from typing import Any, Dict, Optional
+from typing import TYPE_CHECKING, Any, Dict, Optional
 
 import torch
 from transformers import AutoConfig
 
 
+if TYPE_CHECKING:
+    from compressed_tensors.compressors import ModelCompressor
+
+
 __all__ = [
     "infer_compressor_from_model_config",
     "fix_fsdp_module_name",