diff --git a/tests/integration/test_model/test_model_recipe.py b/tests/integration/test_model/test_model_recipe.py index 5c4f32e..95084a5 100644 --- a/tests/integration/test_model/test_model_recipe.py +++ b/tests/integration/test_model/test_model_recipe.py @@ -146,12 +146,11 @@ def test_custom_args(self): job.algorithm_spec.code_dir.location_value.bucket, session.oss_bucket.bucket_name, ) - + def test_compression(self): model = RegisteredModel(model_name="qwen2-0.5b-instruct", model_provider="pai") compress_recipe = model.model_recipe( - recipe_type=ModelRecipeType.COMPRESSION, - method='Quantization:MinMax-8Bit' + recipe_type=ModelRecipeType.COMPRESSION, method="Quantization:MinMax-8Bit" ) compress_recipe.run() self.assertIsNotNone(compress_recipe.latest_job.output_path())