From 36bfa6312a99b83e5f3f9d22a1f01e0d7162179e Mon Sep 17 00:00:00 2001
From: Ian Liao <55819364+ian-Liaozy@users.noreply.github.com>
Date: Fri, 31 Jul 2026 02:09:53 +0000
Subject: [PATCH] Fix flaky unit test to pass post-submit checks

---
 .../runners/interactive/recording_manager_test.py            | 5 ++++-
 1 file changed, 4 insertions(+), 1 deletion(-)

diff --git a/sdks/python/apache_beam/runners/interactive/recording_manager_test.py b/sdks/python/apache_beam/runners/interactive/recording_manager_test.py
index 55c3bd91cfd6..b43d8aeb8731 100644
--- a/sdks/python/apache_beam/runners/interactive/recording_manager_test.py
+++ b/sdks/python/apache_beam/runners/interactive/recording_manager_test.py
@@ -1139,7 +1139,10 @@ def test_get_pipeline_graph_not_cached(self):
         t.unique_name
         for t in graph2._pipeline_proto.components.transforms.values()
     ]
-    self.assertIn('Map1', transform_names)
+    self.assertTrue(
+        any('Map1' in name for name in transform_names),
+        f"Expected 'Map1' in one of the transform names, got: {transform_names}"
+    )
 
   def test_wait_for_completion_raises_exception_on_failure(self):
     future = Future()