diff --git a/homeassistant/components/osramlightify/light.py b/homeassistant/components/osramlightify/light.py index 20e50b8b7fd..a247497550f 100644 --- a/homeassistant/components/osramlightify/light.py +++ b/homeassistant/components/osramlightify/light.py @@ -18,9 +18,8 @@ from homeassistant.components.light import ( SUPPORT_BRIGHTNESS, SUPPORT_COLOR, SUPPORT_COLOR_TEMP, - SUPPORT_EFFECT, - SUPPORT_TRANSITION, LightEntity, + LightEntityFeature, ) from homeassistant.const import CONF_HOST from homeassistant.core import HomeAssistant @@ -210,13 +209,18 @@ class Luminary(LightEntity): """Get list of supported features.""" features = 0 if "lum" in self._luminary.supported_features(): - features = features | SUPPORT_BRIGHTNESS | SUPPORT_TRANSITION + features = features | SUPPORT_BRIGHTNESS | LightEntityFeature.TRANSITION if "temp" in self._luminary.supported_features(): - features = features | SUPPORT_COLOR_TEMP | SUPPORT_TRANSITION + features = features | SUPPORT_COLOR_TEMP | LightEntityFeature.TRANSITION if "rgb" in self._luminary.supported_features(): - features = features | SUPPORT_COLOR | SUPPORT_TRANSITION | SUPPORT_EFFECT + features = ( + features + | SUPPORT_COLOR + | LightEntityFeature.TRANSITION + | LightEntityFeature.EFFECT + ) return features @@ -416,7 +420,7 @@ class OsramLightifyGroup(Luminary): """Get list of supported features.""" features = super()._get_supported_features() if self._luminary.scenes(): - features = features | SUPPORT_EFFECT + features = features | LightEntityFeature.EFFECT return features