From 86f96db9b01ee671b18d1978e70c7dd7caa3fd38 Mon Sep 17 00:00:00 2001 From: Marc Mueller <30130371+cdce8p@users.noreply.github.com> Date: Thu, 2 May 2024 00:00:30 +0200 Subject: [PATCH] Improve asuswrt decorator typing (#116563) --- homeassistant/components/asuswrt/bridge.py | 6 ++++-- 1 file changed, 4 insertions(+), 2 deletions(-) diff --git a/homeassistant/components/asuswrt/bridge.py b/homeassistant/components/asuswrt/bridge.py index c177fb1bb20..579f894ff61 100644 --- a/homeassistant/components/asuswrt/bridge.py +++ b/homeassistant/components/asuswrt/bridge.py @@ -66,10 +66,12 @@ _ReturnFuncType = Callable[[_AsusWrtBridgeT], Coroutine[Any, Any, dict[str, Any] def handle_errors_and_zip( exceptions: type[Exception] | tuple[type[Exception], ...], keys: list[str] | None -) -> Callable[[_FuncType], _ReturnFuncType]: +) -> Callable[[_FuncType[_AsusWrtBridgeT]], _ReturnFuncType[_AsusWrtBridgeT]]: """Run library methods and zip results or manage exceptions.""" - def _handle_errors_and_zip(func: _FuncType) -> _ReturnFuncType: + def _handle_errors_and_zip( + func: _FuncType[_AsusWrtBridgeT], + ) -> _ReturnFuncType[_AsusWrtBridgeT]: """Run library methods and zip results or manage exceptions.""" @functools.wraps(func)