Remove entity description mixin in Trafikverket Train (#112956)

This commit is contained in:
Joost Lekkerkerker 2024-03-10 18:15:18 +01:00 committed by GitHub
parent 13b8fd4149
commit 39c617eee6
No known key found for this signature in database
GPG Key ID: B5690EEEBB952194

View File

@ -26,20 +26,13 @@ from .coordinator import TrainData, TVDataUpdateCoordinator
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",