fix use of asyncio_timeout

NabuCasa · Nov 24, 2024 · db21b70 · db21b70
1 parent b90ca2b
commit db21b70
Show file tree

Hide file tree

Showing 2 changed files with 3 additions and 4 deletions.
diff --git a/universal_silabs_flasher/flasher.py b/universal_silabs_flasher/flasher.py
@@ -284,7 +284,7 @@ async def enter_bootloader(self) -> None:
                     await spinel.enter_bootloader()
         elif self.app_type is ApplicationType.ROUTER:
             async with self._connect_router(self.app_baudrate) as router:
-                async with async_timeout.timeout(PROBE_TIMEOUT):
+                async with asyncio_timeout(PROBE_TIMEOUT):
                     await router.enter_bootloader()
         elif self.app_type is ApplicationType.EZSP:
             async with self._connect_ezsp(self.app_baudrate) as ezsp:

diff --git a/universal_silabs_flasher/router.py b/universal_silabs_flasher/router.py
@@ -5,10 +5,9 @@
 import logging
 import re
 
-import async_timeout
 from zigpy.serial import SerialProtocol
 
-from .common import PROBE_TIMEOUT, StateMachine, Version
+from .common import PROBE_TIMEOUT, StateMachine, Version, asyncio_timeout
 
 _LOGGER = logging.getLogger(__name__)
 
@@ -38,7 +37,7 @@ def __init__(self) -> None:
 
     async def probe(self) -> Version:
         """Attempt to communicate with the router."""
-        async with async_timeout.timeout(PROBE_TIMEOUT):
+        async with asyncio_timeout(PROBE_TIMEOUT):
             return await self.router_info()
 
     async def router_info(self) -> Version: