fix lint error

aliyun · Oct 21, 2024 · db6e229 · db6e229
1 parent 5c3d0e0
commit db6e229
Showing 1 changed file with 2 additions and 3 deletions.
diff --git 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())