Skip to content

Commit c4e6066

Browse files
zoyahavtfx-copybara
authored andcommitted
Update error regex in test to support a slightly different version of the expected error.
PiperOrigin-RevId: 597865655
1 parent bab1a8f commit c4e6066

File tree

1 file changed

+2
-2
lines changed

1 file changed

+2
-2
lines changed

tensorflow_transform/beam/impl_test.py

Lines changed: 2 additions & 2 deletions
Original file line numberDiff line numberDiff line change
@@ -4310,8 +4310,8 @@ def preprocessing_fn(inputs):
43104310
return {'x_scaled': tft.scale_to_0_1(inputs['x'])}
43114311

43124312
# Exception type depends on the runner being used.
4313-
with self.assertRaisesRegexp(
4314-
(RuntimeError, ValueError, TypeError), 'has type .*list'):
4313+
with self.assertRaisesRegex(
4314+
(RuntimeError, ValueError, TypeError), '.*list'):
43154315
# TODO(b/149997088): Remove this explicit use of DirectRunner.
43164316
with beam.Pipeline() as pipeline:
43174317
metadata = tft.DatasetMetadata.from_feature_spec({

0 commit comments

Comments
 (0)