diff --git a/homeassistant/components/switcher_kis/button.py b/homeassistant/components/switcher_kis/button.py index 093077ed818..b0e45f1374a 100644 --- a/homeassistant/components/switcher_kis/button.py +++ b/homeassistant/components/switcher_kis/button.py @@ -30,21 +30,14 @@ from .const import SIGNAL_DEVICE_ADD from .utils import get_breeze_remote_manager -@dataclass(frozen=True) -class SwitcherThermostatButtonDescriptionMixin: - """Mixin to describe a Switcher Thermostat Button entity.""" +@dataclass(frozen=True, kw_only=True) +class SwitcherThermostatButtonEntityDescription(ButtonEntityDescription): + """Class to describe a Switcher Thermostat Button entity.""" press_fn: Callable[[SwitcherType2Api, SwitcherBreezeRemote], SwitcherBaseResponse] supported: Callable[[SwitcherBreezeRemote], bool] -@dataclass(frozen=True) -class SwitcherThermostatButtonEntityDescription( - ButtonEntityDescription, SwitcherThermostatButtonDescriptionMixin -): - """Class to describe a Switcher Thermostat Button entity.""" - - THERMOSTAT_BUTTONS = [ SwitcherThermostatButtonEntityDescription( key="assume_on",