1
mirror of https://github.com/home-assistant/core synced 2024-08-02 23:40:32 +02:00

Use ColorMode enum in modbus (#70522)

This commit is contained in:
epenet 2022-04-23 21:24:06 +02:00 committed by GitHub
parent e40b4bee2d
commit a2bf33e74c
No known key found for this signature in database
GPG Key ID: 4AEE18F83AFDEB23

View File

@ -3,7 +3,7 @@ from __future__ import annotations
from typing import Any
from homeassistant.components.light import COLOR_MODE_ONOFF, LightEntity
from homeassistant.components.light import ColorMode, LightEntity
from homeassistant.const import CONF_LIGHTS, CONF_NAME
from homeassistant.core import HomeAssistant
from homeassistant.helpers.entity_platform import AddEntitiesCallback
@ -36,8 +36,8 @@ async def async_setup_platform(
class ModbusLight(BaseSwitch, LightEntity):
"""Class representing a Modbus light."""
_attr_color_mode = COLOR_MODE_ONOFF
_attr_supported_color_modes = {COLOR_MODE_ONOFF}
_attr_color_mode = ColorMode.ONOFF
_attr_supported_color_modes = {ColorMode.ONOFF}
async def async_turn_on(self, **kwargs: Any) -> None:
"""Set light on."""