diff --git a/tests/python/frontend/tflite/test_forward.py b/tests/python/frontend/tflite/test_forward.py index 9b09eea1d984f..686562ef97e4e 100644 --- a/tests/python/frontend/tflite/test_forward.py +++ b/tests/python/frontend/tflite/test_forward.py @@ -1591,7 +1591,8 @@ def representative_data_gen(): def test_forward_quantize_dequantize(): """ Quantize Dequantize """ data = np.random.uniform(0, 1, (1, 4, 4, 3)).astype("float32") - _test_quantize_dequantize(data) + if package_version.parse(tf.VERSION) >= package_version.parse('2.0.0'): + _test_quantize_dequantize(data) #######################################################################