Fix AbcTest.testAbstract assertion

This commit is contained in:
Junha Park 2021-03-02 13:16:03 -08:00
parent b6f149d84c
commit f350faf71b

View File

@ -335,8 +335,7 @@ class ForwardMethodsTest(test_util.TensorFlowTestCase):
class AbcTest(test_util.TensorFlowTestCase):
def testAbstract(self):
msg = "Can't instantiate .* abstract methods"
with self.assertRaisesRegex(TypeError, msg):
with self.assertRaises(TypeError):
AbstractModule() # pylint: disable=abstract-class-instantiated
def testConcrete(self):