Style

Signed-off-by: Rahul Tuli <[email protected]>

Author: rahul-tuli

src/llmcompressor/recipe/recipe.py

@@ -533,11 +533,12 @@ def model_dump(self, *args, **kwargs) -> Dict[str, Any]:
         expected by the validation logic, ensuring round-trip compatibility with
         model_validate.
 
-        :return: A dictionary representation of the recipe compatible with model_validate
+        :return: A dictionary representation of the recipe compatible with
+            model_validate
         """
         # Get the base dictionary from parent class
         base_dict = super().model_dump(*args, **kwargs)
-        
+
         # Transform stages into the expected format
         if "stages" in base_dict:
             stages_dict = {}

tests/recipe/test_recipe.py

@@ -1,5 +1,6 @@
 from src.llmcompressor.recipe import Recipe
 
+
 def test_recipe_model_dump():
     """Test that model_dump produces a format compatible with model_validate."""
     # Create a recipe with multiple stages and modifiers