Merge pull request #48 from PandABlocks/fix-deprecation-warning

Create a task to pass to asyncio.wait
PandABlocks · Sep 1, 2023 · f99c713 · f99c713
2 parents e24fdfc + ff39a7b
commit f99c713
Showing 1 changed file with 2 additions and 1 deletion.
diff --git a/pandablocks/asyncio.py b/pandablocks/asyncio.py
@@ -32,7 +32,8 @@ async def write_and_drain(self, data: bytes, timeout: Optional[float] = None):
 
         # Cannot simply await the drain, as if the remote end has disconnected
         # then the drain will never complete as the OS cannot clear its send buffer.
-        _, pending = await asyncio.wait([writer.drain()], timeout=timeout)
+        write_task = asyncio.create_task(writer.drain())
+        _, pending = await asyncio.wait([write_task], timeout=timeout)
         if len(pending):
             for task in pending:
                 task.cancel()