Use TaskGroups

itamaro · May 3, 2023 · fe365c8 · fe365c8
1 parent 00db076
commit fe365c8
Showing 1 changed file with 6 additions and 3 deletions.
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)