Fix unittest error message assertion

2021-01-11 23:59:15 +01:00 · 2021-01-11 23:59:15 +01:00 · 2239f71ef0
commit 2239f71ef0
parent c4e6c635de
1 changed files with 1 additions and 1 deletions
--- 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)