fix test

Signed-off-by: Kyle Sayers <kylesayrs@gmail.com>

Author: kylesayrs

src/compressed_tensors/compressors/model_compressors/model_compressor.py

@@ -382,8 +382,6 @@ def apply_compression_status(self, model: Module) -> Module:
         def replace_with_compressed(module: Module) -> Module:
             scheme = getattr(module, "quantization_scheme", None)
             if isinstance(module, torch.nn.Linear) and scheme is not None:
-                # compressed_state_dict_2 = self.compress(module)  # debug
-
                 module = CompressedLinear.from_linear(
                     module,
                     quantization_scheme=scheme,

tests/test_compressors/quantized_compressors/test_pack_quant.py

@@ -228,9 +228,9 @@ def test_asymmetric_packed_support(strategy):
     )
 
     compressor = PackedQuantizationCompressor(config=quant_config)
-    quantized_modules_to_args = {"dummy": quant_config.config_groups["group_1"].weights}
+    quantized_modules_to_scheme = {"dummy": quant_config.config_groups["group_1"]}
     compressed_state_dict = compressor.compress(
-        dense_state_dict, names_to_scheme=quantized_modules_to_args
+        dense_state_dict, names_to_scheme=quantized_modules_to_scheme
     )
 
     # compressed state_dict adds one entry for shape