Ensure compatibility with Home Assistant 2025.1

oyvindwe · Dec 21, 2024 · cef7240 · cef7240
1 parent 4867feb
commit cef7240
Showing 1 changed file with 6 additions and 2 deletions.
diff --git a/custom_components/connectlife/water_heater.py b/custom_components/connectlife/water_heater.py
@@ -3,7 +3,6 @@
 
 from homeassistant.components.water_heater import (
     WaterHeaterEntity,
-    WaterHeaterEntityEntityDescription,
     WaterHeaterEntityFeature,
 )
 from homeassistant.config_entries import ConfigEntry
@@ -16,6 +15,7 @@
     UnitOfTemperature,
 )
 from homeassistant.core import HomeAssistant, callback
+from homeassistant.helpers.entity import EntityDescription
 from homeassistant.helpers.entity_platform import AddEntitiesCallback
 
 from .const import (
@@ -36,6 +36,10 @@
 _LOGGER = logging.getLogger(__name__)
 
 
+class WaterHeaterEntityDescription(EntityDescription, frozen_or_thawed=True):
+    """Class to avoid incompatibility with Home Assistant 2025.1."""
+
+
 async def async_setup_entry(
         hass: HomeAssistant,
         config_entry: ConfigEntry,
@@ -88,7 +92,7 @@ def __init__(
         """Initialize the entity."""
         super().__init__(coordinator, appliance, "waterheater", Platform.WATER_HEATER, config_entry)
 
-        self.entity_description = WaterHeaterEntityEntityDescription(
+        self.entity_description = WaterHeaterEntityDescription(
             key=self._attr_unique_id,
             name=appliance.device_nickname,
             translation_key=DOMAIN