🔧 fix parallel example

shroominic · Jun 10, 2024 · d3ca9e2 · d3ca9e2
1 parent da37a00
commit d3ca9e2
Showing 1 changed file with 5 additions and 2 deletions.
diff --git a/examples/parallel.py b/examples/parallel.py
@@ -4,12 +4,15 @@
 
 
 async def main():
- await asyncio.gather(spawn_codebox() for _ in range(10))
+ await asyncio.gather(*(spawn_codebox() for _ in range(10)))
 
 
 async def spawn_codebox():
  async with CodeBox() as codebox:
- print(await codebox.arun("print('Hello World!')"))
+ await codebox.arun("a = 'Hello World!'")
+ a = await codebox.arun("a")
+ assert a == "Hello World!"
+ print("Success!")
 
 
 if __name__ == "__main__":