jeans/core
jeans/core
1
0
Fork 0
mirror of https://github.com/home-assistant/core.git synced 2025-07-15 09:17:10 +00:00
Code Issues Packages Projects Releases Wiki Activity

Improve surepetcare generic typing (#84647)

Browse Source
This commit is contained in:
Marc Mueller 2022-12-27 21:50:33 +01:00 committed by GitHub
parent 1624cb1046
commit cf598bb5fd
No known key found for this signature in database
GPG Key ID: 4AEE18F83AFDEB23
3 changed files with 3 additions and 3 deletions
Show Stats Download Patch File Download Diff File Expand all files Collapse all files

2
homeassistant/components/surepetcare/__init__.py
View File

@ -159,7 +159,7 @@ async def async_unload_entry(hass: HomeAssistant, entry: ConfigEntry) -> bool:
return unload_ok
class SurePetcareDataCoordinator(DataUpdateCoordinator):
class SurePetcareDataCoordinator(DataUpdateCoordinator[dict[int, SurepyEntity]]):
"""Handle Surepetcare data."""
def __init__(self, entry: ConfigEntry, hass: HomeAssistant) -> None:

2
homeassistant/components/surepetcare/entity.py
View File

@ -26,7 +26,7 @@ class SurePetcareEntity(CoordinatorEntity[SurePetcareDataCoordinator]):
self._id = surepetcare_id
surepy_entity: SurepyEntity = coordinator.data[surepetcare_id]
surepy_entity = coordinator.data[surepetcare_id]
if surepy_entity.name:
self._device_name = surepy_entity.name.capitalize()

2
homeassistant/components/surepetcare/sensor.py
View File

@ -96,7 +96,7 @@ class Felaqua(SurePetcareEntity, SensorEntity):
"""Initialize a Sure Petcare Felaqua sensor."""
super().__init__(surepetcare_id, coordinator)
surepy_entity: SurepyFelaqua = coordinator.data[surepetcare_id]
surepy_entity = cast(SurepyFelaqua, coordinator.data[surepetcare_id])
self._attr_name = self._device_name
self._attr_unique_id = self._device_id