diff --git a/tensorflow/lite/testing/model_coverage/model_coverage_lib_test.py b/tensorflow/lite/testing/model_coverage/model_coverage_lib_test.py index 9236181f840..03a0004b2fc 100644 --- a/tensorflow/lite/testing/model_coverage/model_coverage_lib_test.py +++ b/tensorflow/lite/testing/model_coverage/model_coverage_lib_test.py @@ -31,6 +31,7 @@ from tensorflow.python.eager import def_function from tensorflow.python.framework import constant_op from tensorflow.python.framework import dtypes from tensorflow.python.framework import ops +from tensorflow.python.framework import test_util from tensorflow.python.ops import array_ops from tensorflow.python.ops import math_ops from tensorflow.python.platform import test @@ -178,18 +179,21 @@ class EvaluateKerasModel(test.TestCase): os.close(fd) return keras_file + @test_util.run_v1_only('Keras test fails under v2, see b/157266669') def testFloat(self): model = self._getSingleInputKerasModel() keras_file = self._saveKerasModel(model) model_coverage.test_keras_model(keras_file) + @test_util.run_v1_only('Keras test fails under v2, see b/157266669') def testPostTrainingQuantize(self): model = self._getSingleInputKerasModel() keras_file = self._saveKerasModel(model) model_coverage.test_keras_model(keras_file, post_training_quantize=True) + @test_util.run_v1_only('Keras test fails under v2, see b/157266669') def testTargetOps(self): model = self._getSingleInputKerasModel() keras_file = self._saveKerasModel(model)