fix default test case (#193)

* fix default test case * fix
vllm-project · Sep 20, 2024 · 29caf3c · 29caf3c
1 parent 3fb4212
commit 29caf3c
Showing 1 changed file with 2 additions and 6 deletions.
diff --git a/tests/llmcompressor/pytorch/modifiers/pruning/sparsegpt/test_pytorch.py b/tests/llmcompressor/pytorch/modifiers/pruning/sparsegpt/test_pytorch.py
@@ -86,12 +86,8 @@ def test_create_default_quant_modifier(self):
  should_be_default_quant_scheme = modifier.quantization_modifier_.config_groups[
  default_config_group_name
  ]
- self.assertEqual(should_be_default_quant_scheme.input_activations.num_bits, 8)
- # input activations are symmetric by default in QuantizationModifier
- assert should_be_default_quant_scheme.input_activations.symmetric
-
- self.assertEqual(should_be_default_quant_scheme.weights.num_bits, 8)
- assert should_be_default_quant_scheme.weights.symmetric
+ assert should_be_default_quant_scheme.input_activations is None
+ assert should_be_default_quant_scheme.weights is None
 
 
 @pytest.mark.unit