Use typed config entry in fyta (home-assistant#132248)

semyonchetvertnyh · Dec 4, 2024 · cafd209 · cafd209
1 parent 6b7724c
commit cafd209
Show file tree

Hide file tree

Showing 2 changed files with 6 additions and 4 deletions.
diff --git a/homeassistant/components/fyta/__init__.py b/homeassistant/components/fyta/__init__.py
@@ -55,13 +55,15 @@ async def async_setup_entry(hass: HomeAssistant, entry: FytaConfigEntry) -> bool
     return True
 
 
-async def async_unload_entry(hass: HomeAssistant, entry: ConfigEntry) -> bool:
+async def async_unload_entry(hass: HomeAssistant, entry: FytaConfigEntry) -> bool:
     """Unload Fyta entity."""
 
     return await hass.config_entries.async_unload_platforms(entry, PLATFORMS)
 
 
-async def async_migrate_entry(hass: HomeAssistant, config_entry: ConfigEntry) -> bool:
+async def async_migrate_entry(
+    hass: HomeAssistant, config_entry: FytaConfigEntry
+) -> bool:
     """Migrate old entry."""
     _LOGGER.debug("Migrating from version %s", config_entry.version)
 

diff --git a/homeassistant/components/fyta/entity.py b/homeassistant/components/fyta/entity.py
@@ -3,10 +3,10 @@
 from fyta_cli.fyta_models import Plant
 
 from homeassistant.components.sensor import SensorEntityDescription
-from homeassistant.config_entries import ConfigEntry
 from homeassistant.helpers.device_registry import DeviceInfo
 from homeassistant.helpers.update_coordinator import CoordinatorEntity
 
+from . import FytaConfigEntry
 from .const import DOMAIN
 from .coordinator import FytaCoordinator
 
@@ -19,7 +19,7 @@ class FytaPlantEntity(CoordinatorEntity[FytaCoordinator]):
     def __init__(
         self,
         coordinator: FytaCoordinator,
-        entry: ConfigEntry,
+        entry: FytaConfigEntry,
         description: SensorEntityDescription,
         plant_id: int,
     ) -> None: