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: else:
self.loop.run_until_complete(self.disconnected) self.loop.run_until_complete(self.disconnected)
else: 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: if not forever:
tasks.append(self.disconnected) tasks.append(self.disconnected)
self.loop.run_until_complete(asyncio.wait(tasks)) self.loop.run_until_complete(asyncio.wait(tasks))