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
This commit is contained in:
parent
8d021e4030
commit
b8b2dd9609
|
@ -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)
|
||||
|
|
Loading…
Reference in New Issue