From f350faf71b6deaafe71a86ab7d53f17b6a66e01e Mon Sep 17 00:00:00 2001
From: Junha Park <jpark3@scu.edu>
Date: Tue, 2 Mar 2021 13:16:03 -0800
Subject: [PATCH] Fix AbcTest.testAbstract assertion

---
 tensorflow/python/module/module_test.py | 3 +--
 1 file changed, 1 insertion(+), 2 deletions(-)

diff --git a/tensorflow/python/module/module_test.py b/tensorflow/python/module/module_test.py
index 2d1b1627655..9b3816eefc6 100644
--- a/tensorflow/python/module/module_test.py
+++ b/tensorflow/python/module/module_test.py
@@ -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):