Update error message checked in bitcast_op_test

PiperOrigin-RevId: 407098953 Change-Id: Iff698c7756e1c9cfa612feb9030cce5604938690
Marbck · Nov 2, 2021 · 09b3e01 · 09b3e01
1 parent b735112
commit 09b3e01
Showing 1 changed file with 5 additions and 2 deletions.
diff --git a/tensorflow/python/kernel_tests/array_ops/bitcast_op_test.py b/tensorflow/python/kernel_tests/array_ops/bitcast_op_test.py
@@ -61,8 +61,11 @@ def testSameSize(self):
   def testErrors(self):
     x = np.zeros([1, 1], np.int8)
     datatype = dtypes.int32
-    with self.assertRaisesRegex((ValueError, errors.InvalidArgumentError),
-                                "Cannot bitcast from 6 to 3"):
+    # When eager_op_as_function is enabled shape inference will raise
+    # a different more informative error message.
+    with self.assertRaisesRegex(
+        (ValueError, errors.InvalidArgumentError),
+        "Cannot bitcast from 6 to 3|convert from s8.* to S32"):
       array_ops.bitcast(x, datatype, None)
 
   def testEmpty(self):