diff --git a/homeassistant/components/notion/binary_sensor.py b/homeassistant/components/notion/binary_sensor.py index a1c519f228f..8e4d5927152 100644 --- a/homeassistant/components/notion/binary_sensor.py +++ b/homeassistant/components/notion/binary_sensor.py @@ -30,24 +30,17 @@ from .const import ( SENSOR_SMOKE_CO, SENSOR_WINDOW_HINGED, ) -from .model import NotionEntityDescriptionMixin +from .model import NotionEntityDescription -@dataclass(frozen=True) -class NotionBinarySensorDescriptionMixin: - """Define an entity description mixin for binary and regular sensors.""" - - on_state: Literal["alarm", "leak", "low", "not_missing", "open"] - - -@dataclass(frozen=True) +@dataclass(frozen=True, kw_only=True) class NotionBinarySensorDescription( - BinarySensorEntityDescription, - NotionBinarySensorDescriptionMixin, - NotionEntityDescriptionMixin, + BinarySensorEntityDescription, NotionEntityDescription ): """Describe a Notion binary sensor.""" + on_state: Literal["alarm", "leak", "low", "not_missing", "open"] + BINARY_SENSOR_DESCRIPTIONS = ( NotionBinarySensorDescription( diff --git a/homeassistant/components/notion/model.py b/homeassistant/components/notion/model.py index cdfd6e63dad..a774bfdfad3 100644 --- a/homeassistant/components/notion/model.py +++ b/homeassistant/components/notion/model.py @@ -4,8 +4,8 @@ from dataclasses import dataclass from aionotion.sensor.models import ListenerKind -@dataclass(frozen=True) -class NotionEntityDescriptionMixin: - """Define an description mixin Notion entities.""" +@dataclass(frozen=True, kw_only=True) +class NotionEntityDescription: + """Define an description for Notion entities.""" listener_kind: ListenerKind diff --git a/homeassistant/components/notion/sensor.py b/homeassistant/components/notion/sensor.py index 8c4242aec2a..1d2c81addfa 100644 --- a/homeassistant/components/notion/sensor.py +++ b/homeassistant/components/notion/sensor.py @@ -16,11 +16,11 @@ from homeassistant.helpers.entity_platform import AddEntitiesCallback from . import NotionEntity from .const import DOMAIN, SENSOR_MOLD, SENSOR_TEMPERATURE -from .model import NotionEntityDescriptionMixin +from .model import NotionEntityDescription -@dataclass(frozen=True) -class NotionSensorDescription(SensorEntityDescription, NotionEntityDescriptionMixin): +@dataclass(frozen=True, kw_only=True) +class NotionSensorDescription(SensorEntityDescription, NotionEntityDescription): """Describe a Notion sensor."""