Relax the Error type for assertRaises(), as differnet types could be …

…thrown out in eager vs. graph mode Signed-off-by: Yong Tang <yong.tang.github@outlook.com>
yongtang · May 26, 2020 · ffd1a60 · ffd1a60
1 parent be3dee4
commit ffd1a60
Showing 1 changed file with 1 addition and 3 deletions.
diff --git a/tensorflow/python/kernel_tests/random/random_ops_test.py b/tensorflow/python/kernel_tests/random/random_ops_test.py
@@ -416,9 +416,7 @@ def testSingleSessionGraphSeedNotConstant(self):
 
   def testUniformWithInvalidMaxMindShape(self):
     # Test case for GitHub issue 34363.
-    with self.assertRaisesRegexp(
-        errors.InvalidArgumentError,
-        "must be no greater than rank of output shape"):
+    with self.assertRaises((errors.InvalidArgumentError, ValueError)):
       array = array_ops.zeros(shape=(1,))
       random_ops.random_uniform(shape=(), minval=array)