Skip to content

Commit 08dc50a

Browse files
authored
Fix flaky BigQuery persistent retry test (#39539)
1 parent 9019efd commit 08dc50a

1 file changed

Lines changed: 3 additions & 1 deletion

File tree

sdks/python/apache_beam/io/gcp/bigquery_test.py

Lines changed: 3 additions & 1 deletion
Original file line numberDiff line numberDiff line change
@@ -1524,8 +1524,10 @@ def test_insert_rows_json_persistent_retriable_exception(
15241524
# Expecting 1 initial call plus maximum number of retries
15251525
expected_call_count = 1 + bigquery_tools.MAX_RETRIES
15261526

1527+
# This relies on runner-specific mocking behavior which can be
1528+
# inconsistent on Prism.
15271529
with self.assertRaises(Exception) as exc:
1528-
with beam.Pipeline() as p:
1530+
with beam.Pipeline('FnApiRunner') as p:
15291531
_ = (
15301532
p
15311533
| beam.Create([{

0 commit comments

Comments
 (0)