diff --git a/homeassistant/components/trafikverket_weatherstation/sensor.py b/homeassistant/components/trafikverket_weatherstation/sensor.py index 6fd63c83f4e..bd15c34ff01 100644 --- a/homeassistant/components/trafikverket_weatherstation/sensor.py +++ b/homeassistant/components/trafikverket_weatherstation/sensor.py @@ -43,20 +43,13 @@ PRECIPITATION_TYPE = [ ] -@dataclass(frozen=True) -class TrafikverketRequiredKeysMixin: - """Mixin for required keys.""" +@dataclass(frozen=True, kw_only=True) +class TrafikverketSensorEntityDescription(SensorEntityDescription): + """Describes Trafikverket sensor entity.""" value_fn: Callable[[WeatherStationInfo], StateType | datetime] -@dataclass(frozen=True) -class TrafikverketSensorEntityDescription( - SensorEntityDescription, TrafikverketRequiredKeysMixin -): - """Describes Trafikverket sensor entity.""" - - def add_utc_timezone(date_time: datetime | None) -> datetime | None: """Add UTC timezone if datetime.""" if date_time: