Update water_heater.py (#224)

Updated water heater naming to reflect changes in Home Assistant
zxdavb · Jan 5, 2025 · b085d18 · b085d18
1 parent 2bcc11f
commit b085d18
Showing 1 changed file with 2 additions and 2 deletions.
diff --git a/custom_components/ramses_cc/water_heater.py b/custom_components/ramses_cc/water_heater.py
@@ -13,7 +13,7 @@
     STATE_OFF,
     STATE_ON,
     WaterHeaterEntity,
-    WaterHeaterEntityEntityDescription,
+    WaterHeaterEntityDescription,
     WaterHeaterEntityFeature,
 )
 from homeassistant.config_entries import ConfigEntry
@@ -227,7 +227,7 @@ async def async_set_dhw_schedule(self, schedule: str) -> None:
 
 @dataclass(frozen=True, kw_only=True)
 class RamsesWaterHeaterEntityDescription(
-    RamsesEntityDescription, WaterHeaterEntityEntityDescription
+    RamsesEntityDescription, WaterHeaterEntityDescription
 ):
     """Class describing Ramses water heater entities."""