diff --git a/sdks/python/apache_beam/io/gcp/big_query_query_to_table_it_test.py b/sdks/python/apache_beam/io/gcp/big_query_query_to_table_it_test.py index e8cd88842197..052790c4a202 100644 --- a/sdks/python/apache_beam/io/gcp/big_query_query_to_table_it_test.py +++ b/sdks/python/apache_beam/io/gcp/big_query_query_to_table_it_test.py @@ -30,6 +30,8 @@ import pytest from hamcrest.core.core.allof import all_of +from tenacity import retry +from tenacity import stop_after_attempt from apache_beam.io.gcp import big_query_query_to_table_pipeline from apache_beam.io.gcp.bigquery_tools import BigQueryWrapper @@ -155,6 +157,7 @@ def _setup_new_types_env(self): self.assertTrue(passed, 'Error in BQ setup: %s' % errors) @pytest.mark.it_postcommit + @retry(reraise=True, stop=stop_after_attempt(3)) def test_big_query_legacy_sql(self): verify_query = DIALECT_OUTPUT_VERIFY_QUERY % self.output_table expected_checksum = test_utils.compute_hash(DIALECT_OUTPUT_EXPECTED)