From 5be047ac46bdbb6898808a8dc5e4737588d20579 Mon Sep 17 00:00:00 2001 From: Rahul Tuli Date: Mon, 28 Apr 2025 09:18:26 -0400 Subject: [PATCH] Fix: Test Signed-off-by: Rahul Tuli --- tests/e2e/integrations/test_llm_compressor.py | 7 ++----- 1 file changed, 2 insertions(+), 5 deletions(-) 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