diff --git a/tensorflow/python/kernel_tests/cwise_ops_test.py b/tensorflow/python/kernel_tests/cwise_ops_test.py index bec30fafd3a..9dff254463f 100644 --- a/tensorflow/python/kernel_tests/cwise_ops_test.py +++ b/tensorflow/python/kernel_tests/cwise_ops_test.py @@ -219,7 +219,8 @@ class ComparisonOpTest(test.TestCase): with self.subTest(t=t, f=f): with self.assertRaisesRegex( (ValueError, errors.InvalidArgumentError), - "Incompatible shapes|Dimensions must be equal"): + "Incompatible shapes|Dimensions must be equal|" + "required broadcastable shapes"): f(x.astype(t), y.astype(t))