Fix: Warning about unsupported color modes

JurajNyiri · Jul 4, 2024 · b58083d · b58083d
1 parent a0d25e5
commit b58083d
Showing 1 changed file with 4 additions and 0 deletions.
diff --git a/custom_components/lights_app/entities.py b/custom_components/lights_app/entities.py
@@ -4,6 +4,7 @@
 from homeassistant.components.light import LightEntity
 from homeassistant.components.switch import SwitchEntity
 from homeassistant.components.number import NumberEntity
+from homeassistant.components.light import ColorMode
 from .const import DOMAIN
 
 
@@ -20,6 +21,9 @@ def __init__(self, hass, config_entry: dict, entryData: dict, name_suffix: str):
         self._enabled = False
         self._name = "Lights App"
         self._name_suffix = name_suffix
+
+        self._attr_color_mode = ColorMode.ONOFF
+        self._attr_supported_color_modes = set([ColorMode.ONOFF])
         Entity.__init__(self)
 
     @property