diff --git a/homeassistant/components/tradfri/base_class.py b/homeassistant/components/tradfri/base_class.py index bd1968dfd15..a2b7304cc3e 100644 --- a/homeassistant/components/tradfri/base_class.py +++ b/homeassistant/components/tradfri/base_class.py @@ -2,7 +2,7 @@ from __future__ import annotations from abc import abstractmethod -from collections.abc import Callable +from collections.abc import Callable, Coroutine from functools import wraps from typing import Any, cast @@ -20,7 +20,7 @@ from .coordinator import TradfriDeviceDataUpdateCoordinator def handle_error( func: Callable[[Command | list[Command]], Any] -) -> Callable[[str], Any]: +) -> Callable[[Command | list[Command]], Coroutine[Any, Any, None]]: """Handle tradfri api call error.""" @wraps(func)