From b8b2dd9609489d237cc5299885744a5855cd3d26 Mon Sep 17 00:00:00 2001 From: Jared Duke Date: Thu, 21 May 2020 17:16:30 -0700 Subject: [PATCH] Temporarily blacklist keras model_coverage_lib_tests from TF2 testing This target appears to have never been executed under TF2 conditions, and the Keras-specific tests are now failing. PiperOrigin-RevId: 312770146 Change-Id: Icfc2ac6c7c73dda1db2c29a0a022d2ea8ea9c0da --- .../lite/testing/model_coverage/model_coverage_lib_test.py | 4 ++++ 1 file changed, 4 insertions(+) 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)