migrate to wait_for_upload_aiotasks for aiotasks cleanup (#1353)

This commit is contained in:
Shuchang Zheng 2024-12-07 18:13:53 -08:00 committed by GitHub
parent 620b5bfdea
commit e20de1e734
No known key found for this signature in database
GPG key ID: B5690EEEBB952194
3 changed files with 11 additions and 31 deletions

View file

@ -859,7 +859,7 @@ class WorkflowService:
)
LOG.info("Persisted browser session for workflow run", workflow_run_id=workflow_run.workflow_run_id)
await app.ARTIFACT_MANAGER.wait_for_upload_aiotasks_for_tasks(all_workflow_task_ids)
await app.ARTIFACT_MANAGER.wait_for_upload_aiotasks(all_workflow_task_ids)
try:
async with asyncio.timeout(SAVE_DOWNLOADED_FILES_TIMEOUT):