diff --git a/homeassistant/components/omnilogic/common.py b/homeassistant/components/omnilogic/common.py index e28fd53d6fe..4ef78477afe 100644 --- a/homeassistant/components/omnilogic/common.py +++ b/homeassistant/components/omnilogic/common.py @@ -20,7 +20,7 @@ from .const import ALL_ITEM_KINDS, DOMAIN _LOGGER = logging.getLogger(__name__) -class OmniLogicUpdateCoordinator(DataUpdateCoordinator): +class OmniLogicUpdateCoordinator(DataUpdateCoordinator[dict[tuple, dict[str, Any]]]): """Class to manage fetching update data from single endpoint.""" def __init__( diff --git a/homeassistant/components/omnilogic/sensor.py b/homeassistant/components/omnilogic/sensor.py index 23cff2382f6..5cb7605b854 100644 --- a/homeassistant/components/omnilogic/sensor.py +++ b/homeassistant/components/omnilogic/sensor.py @@ -23,7 +23,9 @@ async def async_setup_entry( ) -> None: """Set up the sensor platform.""" - coordinator = hass.data[DOMAIN][entry.entry_id][COORDINATOR] + coordinator: OmniLogicUpdateCoordinator = hass.data[DOMAIN][entry.entry_id][ + COORDINATOR + ] entities = [] for item_id, item in coordinator.data.items(): diff --git a/homeassistant/components/omnilogic/switch.py b/homeassistant/components/omnilogic/switch.py index 2d2ad08d38a..5065007d97e 100644 --- a/homeassistant/components/omnilogic/switch.py +++ b/homeassistant/components/omnilogic/switch.py @@ -23,7 +23,9 @@ async def async_setup_entry( ) -> None: """Set up the light platform.""" - coordinator = hass.data[DOMAIN][entry.entry_id][COORDINATOR] + coordinator: OmniLogicUpdateCoordinator = hass.data[DOMAIN][entry.entry_id][ + COORDINATOR + ] entities = [] for item_id, item in coordinator.data.items():