Improve omnilogic generic typing (#84686)

This commit is contained in:
Marc Mueller 2022-12-28 11:10:49 +01:00 committed by GitHub
parent 5e04a87cc6
commit 059ecd10a2
No known key found for this signature in database
GPG Key ID: 4AEE18F83AFDEB23
3 changed files with 7 additions and 3 deletions

View File

@ -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__(

View File

@ -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():

View File

@ -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():