From 39c617eee6626646d5b436f7d8a438775a0394d9 Mon Sep 17 00:00:00 2001 From: Joost Lekkerkerker Date: Sun, 10 Mar 2024 18:15:18 +0100 Subject: [PATCH] Remove entity description mixin in Trafikverket Train (#112956) --- .../components/trafikverket_train/sensor.py | 13 +++---------- 1 file changed, 3 insertions(+), 10 deletions(-) diff --git a/homeassistant/components/trafikverket_train/sensor.py b/homeassistant/components/trafikverket_train/sensor.py index adc8250b8d9109..22d8aba47254d3 100644 --- a/homeassistant/components/trafikverket_train/sensor.py +++ b/homeassistant/components/trafikverket_train/sensor.py @@ -26,20 +26,13 @@ ATTR_PRODUCT_FILTER = "product_filter" -@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[[TrainData], StateType | datetime] -@dataclass(frozen=True) -class TrafikverketSensorEntityDescription( - SensorEntityDescription, TrafikverketRequiredKeysMixin -): - """Describes Trafikverket sensor entity.""" - - SENSOR_TYPES: tuple[TrafikverketSensorEntityDescription, ...] = ( TrafikverketSensorEntityDescription( key="departure_time",