cleanup of deprecated test methods

PiperOrigin-RevId: 716071485
tensorflow · Jan 16, 2025 · 7912cc1 · 7912cc1
1 parent c60e42e
commit 7912cc1
Show file tree

Hide file tree

Showing 2 changed files with 3 additions and 3 deletions.
diff --git a/tensorflow_recommenders/layers/feature_interaction/dcn_test.py b/tensorflow_recommenders/layers/feature_interaction/dcn_test.py
@@ -50,15 +50,15 @@ def test_one_input(self):
     self.assertAllClose(np.asarray([[0.16, 0.32, 0.48]]), output)
 
   def test_unsupported_input_dim(self):
-    with self.assertRaisesRegexp(ValueError,
+    with self.assertRaisesRegex(ValueError,
                                  r"dimension mismatch"):
       x0 = np.random.random((12, 5))
       x = np.random.random((12, 7))
       layer = Cross()
       layer(x0, x)
 
   def test_invalid_diag_scale(self):
-    with self.assertRaisesRegexp(ValueError,
+    with self.assertRaisesRegex(ValueError,
                                  r"`diag_scale` should be non-negative"):
       x0 = np.asarray([[0.1, 0.2, 0.3]]).astype(np.float32)
       x = np.asarray([[0.4, 0.5, 0.6]]).astype(np.float32)

diff --git a/tensorflow_recommenders/layers/feature_interaction/dot_interaction_test.py b/tensorflow_recommenders/layers/feature_interaction/dot_interaction_test.py
@@ -64,7 +64,7 @@ def test_valid_input(self):
                                      f13, f23, 0]]), output)
 
   def test_non_matching_dimensions(self):
-    with self.assertRaisesRegexp(ValueError, r"dimensions must be equal"):
+    with self.assertRaisesRegex(ValueError, r"dimensions must be equal"):
       feature1 = np.asarray([[0.1, 0.2, 0.3]]).astype(np.float32)
       feature2 = np.asarray([[2.0, -1.0, 1.0]]).astype(np.float32)
       feature3 = np.asarray([[0.0, 1.0]]).astype(np.float32)