align tests to new MCTQ release

sony · Apr 11, 2024 · f6a728d · f6a728d
1 parent c52b994
commit f6a728d
Show file tree

Hide file tree

Showing 2 changed files with 4 additions and 8 deletions.
diff --git a/tests/trainable_infrastructure_tests/keras/trainable_keras/test_keras_base_quantizer.py b/tests/trainable_infrastructure_tests/keras/trainable_keras/test_keras_base_quantizer.py
@@ -40,9 +40,8 @@ def get_weights_quantization_config(self):
     def run_test(self):
         with self.unit_test.assertRaises(Exception) as e:
             ZeroWeightsQuantizer(self.get_weights_quantization_config())
-            # TODO: In next MCTQ release, this test will fail due to KMEANS removal. Fix: "QuantizationMethod.SYMMETRIC: 3" -> "QuantizationMethod.SYMMETRIC: 2"
         self.unit_test.assertEqual(f'Quantization method mismatch. Expected methods: [<QuantizationMethod.POWER_OF_TWO: 0>, '
-                                   f''f'<QuantizationMethod.SYMMETRIC: 3>], received: QuantizationMethod.UNIFORM.',
+                                   f'<QuantizationMethod.SYMMETRIC: 2>], received: QuantizationMethod.UNIFORM.',
                                    str(e.exception))
 
         with self.unit_test.assertRaises(Exception) as e:
@@ -73,9 +72,8 @@ def get_activation_quantization_config(self):
     def run_test(self):
         with self.unit_test.assertRaises(Exception) as e:
             ZeroActivationsQuantizer(self.get_activation_quantization_config())
-        # TODO: In next MCTQ release, this test will fail due to KMEANS removal. Fix: "QuantizationMethod.SYMMETRIC: 3" -> "QuantizationMethod.SYMMETRIC: 2"
         self.unit_test.assertEqual(f'Quantization method mismatch. Expected methods: [<QuantizationMethod.POWER_OF_TWO: 0>, '
-                                   f'<QuantizationMethod.SYMMETRIC: 3>], received: QuantizationMethod.UNIFORM.',
+                                   f'<QuantizationMethod.SYMMETRIC: 2>], received: QuantizationMethod.UNIFORM.',
                                    str(e.exception))
 
         with self.unit_test.assertRaises(Exception) as e:

diff --git a/...s/trainable_infrastructure_tests/pytorch/trainable_pytorch/test_pytorch_base_quantizer.py b/...s/trainable_infrastructure_tests/pytorch/trainable_pytorch/test_pytorch_base_quantizer.py
@@ -39,8 +39,7 @@ def run_test(self):
 
         with self.unit_test.assertRaises(Exception) as e:
             ZeroWeightsQuantizer(self.get_weights_quantization_config())
-        # TODO: In next MCTQ release, this test will fail due to KMEANS removal. Fix: "QuantizationMethod.SYMMETRIC: 3" -> "QuantizationMethod.SYMMETRIC: 2"
-        self.unit_test.assertEqual(f'Quantization method mismatch. Expected methods: [<QuantizationMethod.POWER_OF_TWO: 0>, <QuantizationMethod.SYMMETRIC: 3>], received: QuantizationMethod.UNIFORM.', str(e.exception))
+        self.unit_test.assertEqual(f'Quantization method mismatch. Expected methods: [<QuantizationMethod.POWER_OF_TWO: 0>, <QuantizationMethod.SYMMETRIC: 2>], received: QuantizationMethod.UNIFORM.', str(e.exception))
 
         with self.unit_test.assertRaises(Exception) as e:
             ZeroWeightsQuantizer(self.get_activation_quantization_config())
@@ -67,8 +66,7 @@ def run_test(self):
 
         with self.unit_test.assertRaises(Exception) as e:
             ZeroActivationsQuantizer(self.get_activation_quantization_config())
-        # TODO: In next MCTQ release, this test will fail due to KMEANS removal. Fix: "QuantizationMethod.SYMMETRIC: 3" -> "QuantizationMethod.SYMMETRIC: 2"
-        self.unit_test.assertEqual(f'Quantization method mismatch. Expected methods: [<QuantizationMethod.POWER_OF_TWO: 0>, <QuantizationMethod.SYMMETRIC: 3>], received: QuantizationMethod.UNIFORM.', str(e.exception))
+        self.unit_test.assertEqual(f'Quantization method mismatch. Expected methods: [<QuantizationMethod.POWER_OF_TWO: 0>, <QuantizationMethod.SYMMETRIC: 2>], received: QuantizationMethod.UNIFORM.', str(e.exception))
 
         with self.unit_test.assertRaises(Exception) as e:
             ZeroActivationsQuantizer(self.get_weights_quantization_config())