Skip to content

Commit 459b7ee

Browse files
authored
Fix flaky unit test to pass post-submit checks (#39562)
1 parent 9c561e2 commit 459b7ee

1 file changed

Lines changed: 4 additions & 1 deletion

File tree

sdks/python/apache_beam/runners/interactive/recording_manager_test.py

Lines changed: 4 additions & 1 deletion
Original file line numberDiff line numberDiff line change
@@ -1139,7 +1139,10 @@ def test_get_pipeline_graph_not_cached(self):
11391139
t.unique_name
11401140
for t in graph2._pipeline_proto.components.transforms.values()
11411141
]
1142-
self.assertIn('Map1', transform_names)
1142+
self.assertTrue(
1143+
any('Map1' in name for name in transform_names),
1144+
f"Expected 'Map1' in one of the transform names, got: {transform_names}"
1145+
)
11431146

11441147
def test_wait_for_completion_raises_exception_on_failure(self):
11451148
future = Future()

0 commit comments

Comments
 (0)