From d723d5815ca6b83fac1575bc67d5a5d8feffd72c Mon Sep 17 00:00:00 2001 From: Joost Lekkerkerker Date: Mon, 11 Mar 2024 11:55:08 +0100 Subject: [PATCH] Remove entity description mixin in Transmission (#112958) --- homeassistant/components/transmission/switch.py | 13 +++---------- 1 file changed, 3 insertions(+), 10 deletions(-) diff --git a/homeassistant/components/transmission/switch.py b/homeassistant/components/transmission/switch.py index 1f95fc492fe..8e79d8246e0 100644 --- a/homeassistant/components/transmission/switch.py +++ b/homeassistant/components/transmission/switch.py @@ -18,22 +18,15 @@ from .coordinator import TransmissionDataUpdateCoordinator _LOGGING = logging.getLogger(__name__) -@dataclass(frozen=True) -class TransmissionSwitchEntityDescriptionMixin: - """Mixin for required keys.""" +@dataclass(frozen=True, kw_only=True) +class TransmissionSwitchEntityDescription(SwitchEntityDescription): + """Entity description class for Transmission switches.""" is_on_func: Callable[[TransmissionDataUpdateCoordinator], bool | None] on_func: Callable[[TransmissionDataUpdateCoordinator], None] off_func: Callable[[TransmissionDataUpdateCoordinator], None] -@dataclass(frozen=True) -class TransmissionSwitchEntityDescription( - SwitchEntityDescription, TransmissionSwitchEntityDescriptionMixin -): - """Entity description class for Transmission switches.""" - - SWITCH_TYPES: tuple[TransmissionSwitchEntityDescription, ...] = ( TransmissionSwitchEntityDescription( key="on_off",