Add entity category to Wiz number entities (#76191)

This commit is contained in:
Franck Nijhof 2022-08-04 20:58:12 +02:00 committed by GitHub
parent b2ceb2043b
commit 31d9425e49
No known key found for this signature in database
GPG Key ID: 4AEE18F83AFDEB23

View File

@ -14,6 +14,7 @@ from homeassistant.components.number import (
)
from homeassistant.config_entries import ConfigEntry
from homeassistant.core import HomeAssistant, callback
from homeassistant.helpers.entity import EntityCategory
from homeassistant.helpers.entity_platform import AddEntitiesCallback
from .const import DOMAIN
@ -56,6 +57,7 @@ NUMBERS: tuple[WizNumberEntityDescription, ...] = (
value_fn=lambda device: cast(Optional[int], device.state.get_speed()),
set_value_fn=_async_set_speed,
required_feature="effect",
entity_category=EntityCategory.CONFIG,
),
WizNumberEntityDescription(
key="dual_head_ratio",
@ -67,6 +69,7 @@ NUMBERS: tuple[WizNumberEntityDescription, ...] = (
value_fn=lambda device: cast(Optional[int], device.state.get_ratio()),
set_value_fn=_async_set_ratio,
required_feature="dual_head",
entity_category=EntityCategory.CONFIG,
),
)