From cf598bb5fd8911522d648b4e1089d1b268e4efad Mon Sep 17 00:00:00 2001
From: Marc Mueller <30130371+cdce8p@users.noreply.github.com>
Date: Tue, 27 Dec 2022 21:50:33 +0100
Subject: [PATCH] Improve `surepetcare` generic typing (#84647)

---
 homeassistant/components/surepetcare/__init__.py | 2 +-
 homeassistant/components/surepetcare/entity.py   | 2 +-
 homeassistant/components/surepetcare/sensor.py   | 2 +-
 3 files changed, 3 insertions(+), 3 deletions(-)

diff --git a/homeassistant/components/surepetcare/__init__.py b/homeassistant/components/surepetcare/__init__.py
index 2e6125bc502..ec5d9f63920 100644
--- a/homeassistant/components/surepetcare/__init__.py
+++ b/homeassistant/components/surepetcare/__init__.py
@@ -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:
diff --git a/homeassistant/components/surepetcare/entity.py b/homeassistant/components/surepetcare/entity.py
index 301479c4b95..75d7f4e1c30 100644
--- a/homeassistant/components/surepetcare/entity.py
+++ b/homeassistant/components/surepetcare/entity.py
@@ -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()
diff --git a/homeassistant/components/surepetcare/sensor.py b/homeassistant/components/surepetcare/sensor.py
index 5779fffef50..534f2396751 100644
--- a/homeassistant/components/surepetcare/sensor.py
+++ b/homeassistant/components/surepetcare/sensor.py
@@ -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