Review comment fixed

apache · May 18, 2020 · 84d4b76 · 84d4b76
1 parent 13535b4
commit 84d4b76
Showing 1 changed file with 2 additions and 1 deletion.
diff --git 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)
 
 
 #######################################################################