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__) _LOGGER = logging.getLogger(__name__)
class OmniLogicUpdateCoordinator(DataUpdateCoordinator): class OmniLogicUpdateCoordinator(DataUpdateCoordinator[dict[tuple, dict[str, Any]]]):
"""Class to manage fetching update data from single endpoint.""" """Class to manage fetching update data from single endpoint."""
def __init__( def __init__(

View File

@ -23,7 +23,9 @@ async def async_setup_entry(
) -> None: ) -> None:
"""Set up the sensor platform.""" """Set up the sensor platform."""
coordinator = hass.data[DOMAIN][entry.entry_id][COORDINATOR] coordinator: OmniLogicUpdateCoordinator = hass.data[DOMAIN][entry.entry_id][
COORDINATOR
]
entities = [] entities = []
for item_id, item in coordinator.data.items(): for item_id, item in coordinator.data.items():

View File

@ -23,7 +23,9 @@ async def async_setup_entry(
) -> None: ) -> None:
"""Set up the light platform.""" """Set up the light platform."""
coordinator = hass.data[DOMAIN][entry.entry_id][COORDINATOR] coordinator: OmniLogicUpdateCoordinator = hass.data[DOMAIN][entry.entry_id][
COORDINATOR
]
entities = [] entities = []
for item_id, item in coordinator.data.items(): for item_id, item in coordinator.data.items():