🐛 worker: Remove invalid usage of asyncio.wait with python 3.11

bugbakery · Dec 28, 2023 · a4e451a · a4e451a
1 parent 0804b17
commit a4e451a
Showing 1 changed file with 5 additions and 2 deletions.
diff --git a/worker/transcribee_worker/document.py b/worker/transcribee_worker/document.py
@@ -39,7 +39,10 @@ async def transaction(self, message: str) -> AsyncGenerator:
             await self._send_change(change)
 
     async def _discard_messages(self):
-        pending = [self._stop.wait(), self.conn.recv()]
+        pending = [
+            asyncio.create_task(self._stop.wait()),
+            asyncio.create_task(self.conn.recv()),
+        ]
         while True:
             done, pending = await asyncio.wait(
                 pending, return_when=asyncio.FIRST_COMPLETED
@@ -49,7 +52,7 @@ async def _discard_messages(self):
                     task.cancel()
                     return
             else:
-                pending.add(self.conn.recv())
+                pending.add(asyncio.create_task(self.conn.recv()))
 
     def stop(self):
         self._stop.set()