Skip to content

Commit ab52907

Browse files
tf-transform-teamtfx-copybara
authored andcommitted
Allow TFT unit tests to run on Python 3.12.
PiperOrigin-RevId: 770348527
1 parent b4f910c commit ab52907

File tree

2 files changed

+338
-10
lines changed

2 files changed

+338
-10
lines changed

tensorflow_transform/beam/analysis_graph_builder_test.py

Lines changed: 3 additions & 2 deletions
Original file line numberDiff line numberDiff line change
@@ -50,9 +50,10 @@ def _preprocessing_fn_with_no_analyzers(inputs):
5050
expected_dot_graph_str_tf2=r"""digraph G {
5151
directed=True;
5252
node [shape=Mrecord];
53-
CreateSavedModel [label="{CreateSavedModel|table_initializers: 0|output_signature: OrderedDict([('x_plus_1', \"Tensor\<shape: [None], \<dtype: 'float32'\>\>\")])|label: CreateSavedModel}"];
53+
CreateSavedModel [label="{CreateSavedModel|table_initializers: 0|output_signature: OrderedDict(\{\'x_plus_1\': \"Tensor\\<shape: [None], \\<dtype: 'float32'\\>\\>\"})|label: CreateSavedModel}"];
5454
}
55-
""")
55+
""",
56+
)
5657

5758

5859
def _preprocessing_fn_with_one_analyzer(inputs):

0 commit comments

Comments
 (0)