From f9dbb92e6071f161a3a0ab980e06a79b172877e3 Mon Sep 17 00:00:00 2001
From: epenet <6771947+epenet@users.noreply.github.com>
Date: Sat, 23 Apr 2022 22:10:35 +0200
Subject: [PATCH] Use ColorMode enum in zengge (#70557)

---
 homeassistant/components/zengge/light.py | 13 ++++++-------
 1 file changed, 6 insertions(+), 7 deletions(-)

diff --git a/homeassistant/components/zengge/light.py b/homeassistant/components/zengge/light.py
index 057c643f409..057b049eefd 100644
--- a/homeassistant/components/zengge/light.py
+++ b/homeassistant/components/zengge/light.py
@@ -10,9 +10,8 @@ from homeassistant.components.light import (
     ATTR_BRIGHTNESS,
     ATTR_HS_COLOR,
     ATTR_WHITE,
-    COLOR_MODE_HS,
-    COLOR_MODE_WHITE,
     PLATFORM_SCHEMA,
+    ColorMode,
     LightEntity,
 )
 from homeassistant.const import CONF_DEVICES, CONF_NAME
@@ -100,16 +99,16 @@ class ZenggeLight(LightEntity):
         return self._white
 
     @property
-    def color_mode(self):
+    def color_mode(self) -> ColorMode:
         """Return the current color mode."""
         if self._white != 0:
-            return COLOR_MODE_WHITE
-        return COLOR_MODE_HS
+            return ColorMode.WHITE
+        return ColorMode.HS
 
     @property
-    def supported_color_modes(self):
+    def supported_color_modes(self) -> set[ColorMode | str]:
         """Flag supported color modes."""
-        return {COLOR_MODE_HS, COLOR_MODE_WHITE}
+        return {ColorMode.HS, ColorMode.WHITE}
 
     @property
     def assumed_state(self):