xmlstream: make mypy even happier

This commit is contained in:
mathieui 2024-02-09 22:58:20 +01:00
parent f151f0a7ab
commit 5a5b36ab39

View File

@ -524,7 +524,7 @@ class XMLStream(asyncio.BaseProtocol):
else:
self.loop.run_until_complete(self.disconnected)
else:
tasks: List[asyncio.Task] = [asyncio.Task(asyncio.sleep(timeout))]
tasks: List[Union[asyncio.Task, asyncio.Future]] = [asyncio.Task(asyncio.sleep(timeout))]
if not forever:
tasks.append(self.disconnected)
self.loop.run_until_complete(asyncio.wait(tasks))