Fix turn on / turn off actions

ClusterM · Nov 5, 2024 · e3040e6 · e3040e6
1 parent c1b61c4
commit e3040e6
Showing 1 changed file with 5 additions and 1 deletion.
diff --git a/custom_components/skykettle/water_heater.py b/custom_components/skykettle/water_heater.py
@@ -79,7 +79,11 @@ def entity_category(self):
 
     @property
     def supported_features(self):
-        return WaterHeaterEntityFeature.TARGET_TEMPERATURE | WaterHeaterEntityFeature.OPERATION_MODE
+        return (
+            WaterHeaterEntityFeature.TARGET_TEMPERATURE
+            | WaterHeaterEntityFeature.OPERATION_MODE
+            | WaterHeaterEntityFeature.ON_OFF
+        )
 
     @property
     def temperature_unit(self):