diff --git a/homeassistant/components/trafikverket_ferry/sensor.py b/homeassistant/components/trafikverket_ferry/sensor.py index ff6b9d2e1ca..93f2d1987b6 100644 --- a/homeassistant/components/trafikverket_ferry/sensor.py +++ b/homeassistant/components/trafikverket_ferry/sensor.py @@ -32,21 +32,14 @@ ATTR_OTHER_INFO = "other_info" SCAN_INTERVAL = timedelta(minutes=5) -@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[[dict[str, Any]], StateType | datetime] info_fn: Callable[[dict[str, Any]], StateType | list] | None -@dataclass(frozen=True) -class TrafikverketSensorEntityDescription( - SensorEntityDescription, TrafikverketRequiredKeysMixin -): - """Describes Trafikverket sensor entity.""" - - SENSOR_TYPES: tuple[TrafikverketSensorEntityDescription, ...] = ( TrafikverketSensorEntityDescription( key="departure_time",