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",