set timeout for event wait

aiudirog · Oct 17, 2023 · b1dccb4 · b1dccb4
1 parent f60d2a1
commit b1dccb4
Showing 1 changed file with 3 additions and 3 deletions.
diff --git a/aiuti/asyncio.py b/aiuti/asyncio.py
@@ -393,9 +393,9 @@ async def _wrapper(*args: Any, **kwargs: Any) -> Any:
 
             if waiting:  # Wait for other loop, maybe across threads
                 try:
-                    await aio.wait_for(
-                        ensure_aw(event.wait(), loop),
-                        timeout=30,
+                    await ensure_aw(
+                        aio.wait_for(event.wait(), timeout=30),
+                        loop,
                     )
                 except RuntimeError:  # Target loop most likely closed
                     pass