From 59178780a6d83a485b54ccb273f0ecb47de4698f Mon Sep 17 00:00:00 2001 From: Mohamed Mekkouri <93391238+MekkCyber@users.noreply.github.com> Date: Mon, 23 Dec 2024 16:27:46 +0100 Subject: [PATCH] Fix : VPTQ test (#35394) fix_test --- tests/quantization/vptq_integration/test_vptq.py | 2 +- 1 file changed, 1 insertion(+), 1 deletion(-) diff --git a/tests/quantization/vptq_integration/test_vptq.py b/tests/quantization/vptq_integration/test_vptq.py index faa9a5879d..173afa7d00 100644 --- a/tests/quantization/vptq_integration/test_vptq.py +++ b/tests/quantization/vptq_integration/test_vptq.py @@ -44,7 +44,7 @@ class VptqConfigTest(unittest.TestCase): quantization_config = VptqConfig() vptq_orig_config = quantization_config.to_dict() - self.assertEqual(quantization_config.quant_config, vptq_orig_config["quant_config"]) + self.assertEqual(vptq_orig_config["quant_method"], quantization_config.quant_method) @slow