Use yield to wait transfer tasks

aresnow1 · aresnow1 · commit b09428c8c128 · 2023-01-30T15:21:49.000+08:00
diff --git 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:

Original file line number	Diff line number	Diff line change
`@@ -601,7 +601,7 @@ async def fetch_batch(`
`601`	`601`	`)`
`602`	`602`	`fetch_keys.extend(list(keys))`
`603`	`603`
`604`		`- await asyncio.gather(*transfer_tasks)`
	`604`	`+ yield asyncio.gather(*transfer_tasks)`
`605`	`605`
`606`	`606`	`set_meta_keys = set()`
`607`	`607`	`for data_key in fetch_keys:`