diff --git a/homeassistant/components/rituals_perfume_genie/switch.py b/homeassistant/components/rituals_perfume_genie/switch.py index 6b88f4f26c1..b5828f5ca07 100644 --- a/homeassistant/components/rituals_perfume_genie/switch.py +++ b/homeassistant/components/rituals_perfume_genie/switch.py @@ -18,22 +18,15 @@ from .coordinator import RitualsDataUpdateCoordinator from .entity import DiffuserEntity -@dataclass(frozen=True) -class RitualsEntityDescriptionMixin: - """Mixin values for Rituals entities.""" +@dataclass(frozen=True, kw_only=True) +class RitualsSwitchEntityDescription(SwitchEntityDescription): + """Class describing Rituals switch entities.""" is_on_fn: Callable[[Diffuser], bool] turn_on_fn: Callable[[Diffuser], Awaitable[None]] turn_off_fn: Callable[[Diffuser], Awaitable[None]] -@dataclass(frozen=True) -class RitualsSwitchEntityDescription( - SwitchEntityDescription, RitualsEntityDescriptionMixin -): - """Class describing Rituals switch entities.""" - - ENTITY_DESCRIPTIONS = ( RitualsSwitchEntityDescription( key="is_on",