diff --git a/sdks/python/apache_beam/io/gcp/bigquery_write_it_test.py b/sdks/python/apache_beam/io/gcp/bigquery_write_it_test.py index bb5c36a3e9b9..cd3edf19de5f 100644 --- a/sdks/python/apache_beam/io/gcp/bigquery_write_it_test.py +++ b/sdks/python/apache_beam/io/gcp/bigquery_write_it_test.py @@ -470,7 +470,7 @@ def test_big_query_write_insert_non_transient_api_call_error(self): input_data = [{ 'number': 1, 'str': 'some_string', - }]*500 + }] table_schema = { "fields": [{ @@ -483,7 +483,7 @@ def test_big_query_write_insert_non_transient_api_call_error(self): bq_result_errors = [({ 'number': 1, 'str': 'some_string', - }, "Not Found")]*500 + }, "Not Found")] args = self.test_pipeline.get_full_options_as_args() @@ -491,6 +491,7 @@ def test_big_query_write_insert_non_transient_api_call_error(self): # pylint: disable=expression-not-assigned errors = ( p | 'create' >> beam.Create(input_data) + | beam.WindowInto(beam.transforms.window.FixedWindows(10)) | 'write' >> beam.io.WriteToBigQuery( table_id, schema=table_schema,