Use TaskGroups

itamaro · itamaro · commit fe365c8c3b60 · 2023-05-03T15:16:33.000-07:00
diff --git a/pyperformance/data-files/benchmarks/bm_async_tree/run_benchmark.py b/pyperformance/data-files/benchmarks/bm_async_tree/run_benchmark.py
@@ -52,9 +52,12 @@ async def recurse(self, recurse_level):
             await self.workload_func()
             return
 
-        await asyncio.gather(
-            *[self.recurse(recurse_level - 1) for _ in range(NUM_RECURSE_BRANCHES)]
-        )
+        # await asyncio.gather(
+        #     *[self.recurse(recurse_level - 1) for _ in range(NUM_RECURSE_BRANCHES)]
+        # )
+        async with asyncio.TaskGroup() as tg:
+            for _ in range(NUM_RECURSE_BRANCHES):
+                tg.create_task(self.recurse(recurse_level - 1))
 
     async def run(self):
         await self.recurse(NUM_RECURSE_LEVELS)
