fixing tests to old versions

Summary

Signed-off-by: HDCharles <[email protected]>

Author: HDCharles

tests/llmcompressor/pytorch/modifiers/logarithmic_equalization/test_pytorch.py

@@ -8,18 +8,15 @@
 
 @pytest.mark.unit
 def test_log_equalization_mapping(state):
-    # Use regex patterns with parent-scoped search
-    # Searches for balance layers within the parent of smooth layer
-    mappings = [(["re:^fc2$"], "re:.*block1\\.fc1$")]
+    mappings = [(["seq.fc2"], "seq.block1.fc1")]
     modifier = LogarithmicEqualizationModifier(mappings=mappings)
 
     modifier.ignore = []
     modifier.resolved_mappings_ = modifier._resolve_mappings(state.model)
 
-    assert len(modifier.resolved_mappings_) == 1
+    assert len(modifier.resolved_mappings_) == len(mappings)
 
     mapping = modifier.resolved_mappings_[0]
-    assert mapping.smooth_name == "seq.block1.fc1"
+    assert mapping.smooth_name == mappings[0][1]
     assert isinstance(mapping.smooth_layer, Linear)
-    assert len(mapping.balance_layers) == 1
-    assert isinstance(mapping.balance_layers[0], Linear)
+    assert isinstance(mapping.balance_layers[0], Linear)

tests/llmcompressor/pytorch/modifiers/smoothquant/test_pytorch.py

@@ -4,23 +4,23 @@
 from llmcompressor.modifiers.smoothquant import SmoothQuantModifier
 
 
+import pytest
+from torch.nn import Linear
+
+from llmcompressor.modifiers.smoothquant import SmoothQuantModifier
+
+
 @pytest.mark.unit
 def test_smooth_quant_mapping(state):
-    # Use regex patterns with parent-scoped search
-    # ^fc1$ matches only direct child "fc1", not nested "block1.fc1"
-    mappings = [(["re:^fc1$"], "re:.*fc2$")]
+    mappings = [(["seq.fc1"], "seq.fc2")]
     modifier = SmoothQuantModifier(mappings=mappings)
 
     modifier.ignore = []
     modifier.resolved_mappings_ = modifier._resolve_mappings(state.model)
 
-    # Should match seq.fc2 and block1.fc2 (both end with fc2)
-    assert len(modifier.resolved_mappings_) == 2
+    assert len(modifier.resolved_mappings_) == len(mappings)
 
-    # Verify seq.fc2 mapping - should find only seq.fc1 (direct child)
-    seq_mapping = [
-        m for m in modifier.resolved_mappings_ if m.smooth_name == "seq.fc2"
-    ][0]
-    assert isinstance(seq_mapping.smooth_layer, Linear)
-    assert len(seq_mapping.balance_layers) == 1
-    assert isinstance(seq_mapping.balance_layers[0], Linear)
+    mapping = modifier.resolved_mappings_[0]
+    assert mapping.smooth_name == mappings[0][1]
+    assert isinstance(mapping.smooth_layer, Linear)
+    assert isinstance(mapping.balance_layers[0], Linear)