From eeb2b0f3a97a92a1995845d6e0d4e7267173246d Mon Sep 17 00:00:00 2001 From: Esun Kim Date: Mon, 19 Oct 2020 09:45:12 -0700 Subject: [PATCH] Fix pagination in upload_rbe_results --- tools/run_tests/python_utils/upload_rbe_results.py | 5 +++-- 1 file changed, 3 insertions(+), 2 deletions(-) diff --git a/tools/run_tests/python_utils/upload_rbe_results.py b/tools/run_tests/python_utils/upload_rbe_results.py index 62f57ddc946..93fec98c48b 100755 --- a/tools/run_tests/python_utils/upload_rbe_results.py +++ b/tools/run_tests/python_utils/upload_rbe_results.py @@ -287,7 +287,8 @@ if __name__ == "__main__": if not args.skip_upload: # BigQuery sometimes fails with large uploads, so batch 1,000 rows at a time. - for i in range((len(bq_rows) / 1000) + 1): - _upload_results_to_bq(bq_rows[i * 1000:(i + 1) * 1000]) + MAX_ROWS = 1000 + for i in range(0, len(bq_rows), MAX_ROWS): + _upload_results_to_bq(bq_rows[i:i + MAX_ROWS]) else: print('Skipped upload to bigquery.')