cleanup

Signed-off-by: Brian Dellabetta <bdellabe@redhat.com>

Author: brian-dellabetta

src/compressed_tensors/quantization/lifecycle/apply.py

@@ -148,7 +148,8 @@ def apply_quantization_config(
     ):
         # mark modules to be quantized by adding
         # quant scheme to the matching layers
-        scheme = _scheme_from_targets(target_to_scheme, scheme.targets, name)
+        matched_targets = match_targets(name, submodule, target_to_scheme)
+        scheme = _scheme_from_targets(target_to_scheme, matched_targets, name)
         if (
             run_compressed
             and config.format != CompressionFormat.dense.value

src/compressed_tensors/utils/match.py

@@ -146,11 +146,7 @@ def match_targets(
     targets = sorted(targets, key=lambda x: ("re:" in x, x))
     matched_targets = []
     for target in targets:
-        if _match_name(name, target):
-            matched_targets.append(target)
-
-    for target in targets:
-        if _match_class(module, target) and target not in matched_targets:
+        if _match_name(name, target) or _match_class(module, target):
             matched_targets.append(target)
 
     return matched_targets