[Bugfix] Add fix to preserve modifier order when passed as a list

src/llmcompressor/recipe/recipe.py

@@ -672,7 +672,7 @@ def create_recipe_string_from_modifiers(
             }
         }
     }
-    recipe_str: str = yaml.dump(recipe_dict)
+    recipe_str: str = yaml.dump(recipe_dict, sort_keys=False)
     return recipe_str
 
 

tests/llmcompressor/recipe/test_recipe.py

@@ -3,8 +3,10 @@
 import pytest
 import yaml
 
+from llmcompressor.modifiers import Modifier
 from llmcompressor.modifiers.obcq.base import SparseGPTModifier
 from llmcompressor.recipe import Recipe
+from llmcompressor.recipe.recipe import create_recipe_string_from_modifiers
 from tests.llmcompressor.helpers import valid_recipe_strings
 
 
@@ -96,3 +98,45 @@ def test_recipe_can_be_created_from_modifier_instances():
     ):
         assert isinstance(actual_modifier, type(expected_modifier))
         assert actual_modifier.dict() == expected_modifier.dict()
+
+
+class A_FirstDummyModifier(Modifier):
+    def model_dump(self):
+        return {}
+
+
+class B_SecondDummyModifier(Modifier):
+    def model_dump(self):
+        return {}
+
+
+def test_create_recipe_string_from_modifiers_with_default_group_name():
+    modifiers = [B_SecondDummyModifier(), A_FirstDummyModifier()]
+    expected_recipe_str = (
+        "DEFAULT_stage:\n"
+        "  DEFAULT_modifiers:\n"
+        "    B_SecondDummyModifier: {}\n"
+        "    A_FirstDummyModifier: {}\n"
+    )
+    actual_recipe_str = create_recipe_string_from_modifiers(modifiers)
+    assert actual_recipe_str == expected_recipe_str
+
+
+def test_create_recipe_string_from_modifiers_with_custom_group_name():
+    modifiers = [B_SecondDummyModifier(), A_FirstDummyModifier()]
+    group_name = "custom"
+    expected_recipe_str = (
+        "custom_stage:\n"
+        "  DEFAULT_modifiers:\n"
+        "    B_SecondDummyModifier: {}\n"
+        "    A_FirstDummyModifier: {}\n"
+    )
+    actual_recipe_str = create_recipe_string_from_modifiers(modifiers, group_name)
+    assert actual_recipe_str == expected_recipe_str
+
+
+def test_create_recipe_string_from_modifiers_with_empty_modifiers():
+    modifiers = []
+    expected_recipe_str = "DEFAULT_stage:\n" "  DEFAULT_modifiers: {}\n"
+    actual_recipe_str = create_recipe_string_from_modifiers(modifiers)
+    assert actual_recipe_str == expected_recipe_str

tests/llmcompressor/transformers/obcq/test_mask_structure_preservation.py

@@ -2,7 +2,7 @@
 from pathlib import Path
 
 import pytest
-from compressed_tensors.compressors.utils import tensor_follows_mask_structure
+from compressed_tensors.utils import tensor_follows_mask_structure
 from parameterized import parameterized_class
 
 from llmcompressor.core import reset_session