From b09428c8c1287bd495345f82fb1ae299e3513725 Mon Sep 17 00:00:00 2001 From: aresnow Date: Sat, 28 Jan 2023 11:43:51 +0800 Subject: [PATCH] Use yield to wait transfer tasks --- mars/services/storage/handler.py | 2 +- 1 file changed, 1 insertion(+), 1 deletion(-) diff --git a/mars/services/storage/handler.py b/mars/services/storage/handler.py index 8da758881d..54eba07bd3 100644 --- a/mars/services/storage/handler.py +++ b/mars/services/storage/handler.py @@ -601,7 +601,7 @@ async def fetch_batch( ) fetch_keys.extend(list(keys)) - await asyncio.gather(*transfer_tasks) + yield asyncio.gather(*transfer_tasks) set_meta_keys = set() for data_key in fetch_keys: