diff --git a/homeassistant/components/trafikverket_train/sensor.py b/homeassistant/components/trafikverket_train/sensor.py index 82f248f9de0..7bcd796bb16 100644 --- a/homeassistant/components/trafikverket_train/sensor.py +++ b/homeassistant/components/trafikverket_train/sensor.py @@ -6,7 +6,7 @@ import logging from typing import Any from pytrafikverket import TrafikverketTrain -from pytrafikverket.trafikverket_train import TrainStop +from pytrafikverket.trafikverket_train import StationInfo, TrainStop import voluptuous as vol from homeassistant.components.sensor import ( @@ -166,8 +166,8 @@ class TrainSensor(SensorEntity): self, train_api: TrafikverketTrain, name: str, - from_station: str, - to_station: str, + from_station: StationInfo, + to_station: StationInfo, weekday: list, departuretime: time | None, entry_id: str, @@ -188,7 +188,7 @@ class TrainSensor(SensorEntity): configuration_url="https://api.trafikinfo.trafikverket.se/", ) self._attr_unique_id = create_unique_id( - from_station, to_station, departuretime, weekday + from_station.name, to_station.name, departuretime, weekday ) async def async_update(self) -> None: