From 85b63f4fada400efbae6837d850ce1bdaaa1082c Mon Sep 17 00:00:00 2001
From: bvolpato <brunocvcunha@gmail.com>
Date: Tue, 28 Jul 2026 22:55:02 -0400
Subject: [PATCH] Fix flaky BigQuery persistent retry test

---
 sdks/python/apache_beam/io/gcp/bigquery_test.py | 4 +++-
 1 file changed, 3 insertions(+), 1 deletion(-)

diff --git a/sdks/python/apache_beam/io/gcp/bigquery_test.py b/sdks/python/apache_beam/io/gcp/bigquery_test.py
index 234c99847a44..50d758c6a315 100644
--- a/sdks/python/apache_beam/io/gcp/bigquery_test.py
+++ b/sdks/python/apache_beam/io/gcp/bigquery_test.py
@@ -1524,8 +1524,10 @@ def test_insert_rows_json_persistent_retriable_exception(
     # Expecting 1 initial call plus maximum number of retries
     expected_call_count = 1 + bigquery_tools.MAX_RETRIES
 
+    # This relies on runner-specific mocking behavior which can be
+    # inconsistent on Prism.
     with self.assertRaises(Exception) as exc:
-      with beam.Pipeline() as p:
+      with beam.Pipeline('FnApiRunner') as p:
         _ = (
             p
             | beam.Create([{