From 149cc01ed32b9acbd74c229bdbff7961e2a9a078 Mon Sep 17 00:00:00 2001
From: epenet <6771947+epenet@users.noreply.github.com>
Date: Sat, 23 Apr 2022 21:16:27 +0200
Subject: [PATCH] Use ColorMode enum in iaqualink (#70510)

---
 homeassistant/components/iaqualink/light.py | 9 ++++-----
 1 file changed, 4 insertions(+), 5 deletions(-)

diff --git a/homeassistant/components/iaqualink/light.py b/homeassistant/components/iaqualink/light.py
index 8799f788ae9..505ad8da0cc 100644
--- a/homeassistant/components/iaqualink/light.py
+++ b/homeassistant/components/iaqualink/light.py
@@ -4,9 +4,8 @@ from __future__ import annotations
 from homeassistant.components.light import (
     ATTR_BRIGHTNESS,
     ATTR_EFFECT,
-    COLOR_MODE_BRIGHTNESS,
-    COLOR_MODE_ONOFF,
     DOMAIN,
+    ColorMode,
     LightEntity,
     LightEntityFeature,
 )
@@ -87,11 +86,11 @@ class HassAqualinkLight(AqualinkEntity, LightEntity):
         return list(self.dev.supported_light_effects)
 
     @property
-    def color_mode(self) -> str:
+    def color_mode(self) -> ColorMode:
         """Return the color mode of the light."""
         if self.dev.is_dimmer:
-            return COLOR_MODE_BRIGHTNESS
-        return COLOR_MODE_ONOFF
+            return ColorMode.BRIGHTNESS
+        return ColorMode.ONOFF
 
     @property
     def supported_color_modes(self) -> set[str] | None: