diff --git a/homeassistant/components/tankerkoenig/coordinator.py b/homeassistant/components/tankerkoenig/coordinator.py index 117a58ee2db..4ce9fce7935 100644 --- a/homeassistant/components/tankerkoenig/coordinator.py +++ b/homeassistant/components/tankerkoenig/coordinator.py @@ -31,7 +31,7 @@ _LOGGER = logging.getLogger(__name__) TankerkoenigConfigEntry = ConfigEntry["TankerkoenigDataUpdateCoordinator"] -class TankerkoenigDataUpdateCoordinator(DataUpdateCoordinator): +class TankerkoenigDataUpdateCoordinator(DataUpdateCoordinator[dict[str, PriceInfo]]): """Get the latest data from the API.""" config_entry: TankerkoenigConfigEntry diff --git a/homeassistant/components/tankerkoenig/sensor.py b/homeassistant/components/tankerkoenig/sensor.py index 776ea669d5b..5970f3d3b24 100644 --- a/homeassistant/components/tankerkoenig/sensor.py +++ b/homeassistant/components/tankerkoenig/sensor.py @@ -4,7 +4,7 @@ from __future__ import annotations import logging -from aiotankerkoenig import GasType, PriceInfo, Station +from aiotankerkoenig import GasType, Station from homeassistant.components.sensor import SensorEntity, SensorStateClass from homeassistant.const import ATTR_LATITUDE, ATTR_LONGITUDE, CURRENCY_EURO @@ -109,5 +109,5 @@ class FuelPriceSensor(TankerkoenigCoordinatorEntity, SensorEntity): @property def native_value(self) -> float: """Return the current price for the fuel type.""" - info: PriceInfo = self.coordinator.data[self._station_id] + info = self.coordinator.data[self._station_id] return getattr(info, self._fuel_type)