From 29caf3c73064c8cd1658ea47ded4b72c2156c056 Mon Sep 17 00:00:00 2001 From: Dipika Sikka Date: Fri, 20 Sep 2024 17:26:07 -0400 Subject: [PATCH] fix default test case (#193) * fix default test case * fix --- .../pytorch/modifiers/pruning/sparsegpt/test_pytorch.py | 8 ++------ 1 file changed, 2 insertions(+), 6 deletions(-) diff --git a/tests/llmcompressor/pytorch/modifiers/pruning/sparsegpt/test_pytorch.py b/tests/llmcompressor/pytorch/modifiers/pruning/sparsegpt/test_pytorch.py index 378552316..5421af4cf 100644 --- 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