From 2239f71ef056a6ad18dd4563986fe1e5ce38dce5 Mon Sep 17 00:00:00 2001 From: Lukas Geiger Date: Mon, 11 Jan 2021 23:59:15 +0100 Subject: [PATCH] Fix unittest error message assertion --- tensorflow/python/keras/saving/saved_model/saved_model_test.py | 2 +- 1 file changed, 1 insertion(+), 1 deletion(-) diff --git a/tensorflow/python/keras/saving/saved_model/saved_model_test.py b/tensorflow/python/keras/saving/saved_model/saved_model_test.py index b26ce9f8b23..d685a649f5b 100644 --- a/tensorflow/python/keras/saving/saved_model/saved_model_test.py +++ b/tensorflow/python/keras/saving/saved_model/saved_model_test.py @@ -1161,7 +1161,7 @@ class MetricTest(test.TestCase, parameterized.TestCase): saved_model_dir = self._save_model_dir() tf_save.save(model, saved_model_dir) - with self.assertRaisesRegex(ValueError, 'custom_objects'): + with self.assertRaisesRegex(ValueError, 'metric'): keras_load.load(saved_model_dir) keras_load.load(saved_model_dir, compile=False)