Ensure we hold strong references to tasks

see python/cpython#88831
esphome · Feb 13, 2023 · 0d5019c · 0d5019c
1 parent 6449a47
commit 0d5019c
Show file tree

Hide file tree

Showing 2 changed files with 2 additions and 2 deletions.
diff --git a/aioesphomeapi/connection.py b/aioesphomeapi/connection.py
@@ -158,7 +158,7 @@ def _cleanup(self) -> None:
 
         if self.on_stop and self._connect_complete:
 
-            def _remove_on_stop_task(_fut: asyncio.Future) -> None:
+            def _remove_on_stop_task(_fut: asyncio.Future[None]) -> None:
                 """Remove the stop task from the reconnect loop.
 
                 We need to do this because the asyncio does not hold

diff --git a/aioesphomeapi/reconnect_logic.py b/aioesphomeapi/reconnect_logic.py
@@ -201,7 +201,7 @@ async def stop(self) -> None:
         await self._stop_zc_listen()
 
     def stop_callback(self) -> None:
-        def _remove_stop_task(_fut: asyncio.Future) -> None:
+        def _remove_stop_task(_fut: asyncio.Future[None]) -> None:
             """Remove the stop task from the reconnect loop.
 
             We need to do this because the asyncio does not hold