Skip to content

Commit 29caf3c

Browse files
authored
fix default test case (#193)
* fix default test case * fix
1 parent 3fb4212 commit 29caf3c

File tree

1 file changed

+2
-6
lines changed

1 file changed

+2
-6
lines changed

tests/llmcompressor/pytorch/modifiers/pruning/sparsegpt/test_pytorch.py

Lines changed: 2 additions & 6 deletions
Original file line numberDiff line numberDiff line change
@@ -86,12 +86,8 @@ def test_create_default_quant_modifier(self):
8686
should_be_default_quant_scheme = modifier.quantization_modifier_.config_groups[
8787
default_config_group_name
8888
]
89-
self.assertEqual(should_be_default_quant_scheme.input_activations.num_bits, 8)
90-
# input activations are symmetric by default in QuantizationModifier
91-
assert should_be_default_quant_scheme.input_activations.symmetric
92-
93-
self.assertEqual(should_be_default_quant_scheme.weights.num_bits, 8)
94-
assert should_be_default_quant_scheme.weights.symmetric
89+
assert should_be_default_quant_scheme.input_activations is None
90+
assert should_be_default_quant_scheme.weights is None
9591

9692

9793
@pytest.mark.unit

0 commit comments

Comments
 (0)