diff --git a/tests/e2e/integrations/test_llm_compressor.py b/tests/e2e/integrations/test_llm_compressor.py index 98e8caf49..a3e62c009 100644 --- a/tests/e2e/integrations/test_llm_compressor.py +++ b/tests/e2e/integrations/test_llm_compressor.py @@ -90,12 +90,9 @@ class TestLLMCompressorIntegration: def _check_llmcompressor_model_outputs(temp_dir, save_compressed): - - # recipe.yaml should exist - assert (Path(temp_dir) / "recipe.yaml").exists() - - # sparsity config exists if save_compressed if save_compressed: + assert (Path(temp_dir) / "recipe.yaml").exists() + from compressed_tensors import ModelCompressor from compressed_tensors.config import Sparse24BitMaskConfig