Fix test failures caused by a NumPy 1.21 upgrade
@@ -253,7 +253,7 @@ class TestSavedModelFormatAllModes(keras_parameterized.TestCase):
with previous_losses[0].graph.as_default():
# If we try to compare symbolic Tensors in eager mode assertAllEqual will
# return False even if they are the same Tensor.
- self.assertAllEqual(previous_losses, model.losses)
+ self.assertEqual(previous_losses, model.losses)
if tf.executing_eagerly():
# Test that eager losses are maintained.
