Update error regex in test to support a slightly different version of…

… the expected error. PiperOrigin-RevId: 597865655
tensorflow · Jan 12, 2024 · c4e6066 · c4e6066
1 parent bab1a8f
commit c4e6066
Showing 1 changed file with 2 additions and 2 deletions.
diff --git a/tensorflow_transform/beam/impl_test.py b/tensorflow_transform/beam/impl_test.py
@@ -4310,8 +4310,8 @@ def preprocessing_fn(inputs):
       return {'x_scaled': tft.scale_to_0_1(inputs['x'])}
 
     # Exception type depends on the runner being used.
-    with self.assertRaisesRegexp(
-        (RuntimeError, ValueError, TypeError), 'has type .*list'):
+    with self.assertRaisesRegex(
+        (RuntimeError, ValueError, TypeError), '.*list'):
       # TODO(b/149997088): Remove this explicit use of DirectRunner.
       with beam.Pipeline() as pipeline:
         metadata = tft.DatasetMetadata.from_feature_spec({