diff --git a/homeassistant/components/energyzero/sensor.py b/homeassistant/components/energyzero/sensor.py index 005abb62e91..6438dc6475a 100644 --- a/homeassistant/components/energyzero/sensor.py +++ b/homeassistant/components/energyzero/sensor.py @@ -29,21 +29,14 @@ from .const import DOMAIN, SERVICE_TYPE_DEVICE_NAMES from .coordinator import EnergyZeroData, EnergyZeroDataUpdateCoordinator -@dataclass(frozen=True) -class EnergyZeroSensorEntityDescriptionMixin: - """Mixin for required keys.""" +@dataclass(frozen=True, kw_only=True) +class EnergyZeroSensorEntityDescription(SensorEntityDescription): + """Describes an EnergyZero sensor entity.""" value_fn: Callable[[EnergyZeroData], float | datetime | None] service_type: str -@dataclass(frozen=True) -class EnergyZeroSensorEntityDescription( - SensorEntityDescription, EnergyZeroSensorEntityDescriptionMixin -): - """Describes a Pure Energie sensor entity.""" - - SENSORS: tuple[EnergyZeroSensorEntityDescription, ...] = ( EnergyZeroSensorEntityDescription( key="current_hour_price",