Skip to content

Commit

Permalink
Remove entity description mixin in Trafikverket Train (home-assistant…
Browse files Browse the repository at this point in the history
  • Loading branch information
joostlek authored Mar 10, 2024
1 parent 13b8fd4 commit 39c617e
Showing 1 changed file with 3 additions and 10 deletions.
13 changes: 3 additions & 10 deletions homeassistant/components/trafikverket_train/sensor.py
Original file line number Diff line number Diff line change
Expand Up @@ -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",
Expand Down

0 comments on commit 39c617e

Please sign in to comment.