fix failing tests

Author: vineet-g

tests/quantization/test_eager_quant.py

@@ -3379,7 +3379,7 @@ class TestFP4MLIRExportValidation:
                 torch.randn(1, 32),
                 PerBlockGranularity(axis=1, block_size=32),
                 True,
-                "FP4 activation quantization is not supported for MLIR export",
+                "Core AI export does not support FP4 activation quantization",
                 id="fp4_activation_rejected",
             ),
             pytest.param(

tests/quantization/test_fake_quantize.py

@@ -69,7 +69,6 @@ def test_fake_quant_dequant_no_reduction(qscheme, granularity, qformulation):
     qparam_calculator = StaticQParamsCalculator(range_calculator=range_calculator, **kwargs)
     fq = _DefaultFakeQuantizeImpl(
         qparams_calculator=qparam_calculator,
-        quantization_target=CompressionTargetTensor.WEIGHT,
         **kwargs,
     )
 
@@ -105,7 +104,6 @@ def test_set_granularity():
     qparam_calculator = StaticQParamsCalculator(range_calculator=range_calculator, **kwargs)
     fq = _DefaultFakeQuantizeImpl(
         qparams_calculator=qparam_calculator,
-        quantization_target=CompressionTargetTensor.WEIGHT,
         **kwargs,
     )
     x = torch.randn(2, 5)
@@ -189,7 +187,6 @@ def fq(self, dtype, qscheme, qformulation, granularity):
         qparam_calculator = StaticQParamsCalculator(range_calculator=range_calculator, **kwargs)
         return _DefaultFakeQuantizeImpl(
             qparams_calculator=qparam_calculator,
-            quantization_target=CompressionTargetTensor.WEIGHT,
             **kwargs,
         )
 

tests/quantization/test_graph_mode_quantizer.py

@@ -1369,7 +1369,7 @@ class TestFP4MLIRExportValidation:
                 torch.randn(1, 32),
                 PerBlockGranularity(axis=1, block_size=32),
                 True,
-                "FP4 activation quantization is not supported for MLIR export",
+                "Core AI export does not support FP4 activation quantization",
                 id="fp4_activation_rejected",
             ),
             pytest.param(