Skip to content

Commit 4eeb73c

Browse files
authored
set timeout on calls to bigquery in sync task (#18259)
1 parent 03a6530 commit 4eeb73c

File tree

2 files changed

+2
-1
lines changed

2 files changed

+2
-1
lines changed

tests/unit/packaging/test_tasks.py

Lines changed: 1 addition & 0 deletions
Original file line numberDiff line numberDiff line change
@@ -651,6 +651,7 @@ def find_service(name=None):
651651
"blake2_256_digest": release_file.blake2_256_digest,
652652
},
653653
],
654+
timeout=5,
654655
)
655656
for table in release_files_table.split()
656657
]

warehouse/packaging/tasks.py

Lines changed: 1 addition & 1 deletion
Original file line numberDiff line numberDiff line change
@@ -346,7 +346,7 @@ def update_bigquery_release_files(task, request, dist_metadata):
346346
json_rows[sch.name] = field_data
347347
json_rows = [json_rows]
348348

349-
bq.insert_rows_json(table=table_name, json_rows=json_rows)
349+
bq.insert_rows_json(table=table_name, json_rows=json_rows, timeout=5)
350350

351351

352352
@tasks.task(ignore_result=True, acks_late=True)

0 commit comments

Comments
 (0)