Remove the use of StateType from AccuWeather (#97109)
This commit is contained in:
parent
69d7b035e0
commit
5e88ca23b3
1 changed files with 3 additions and 4 deletions
|
@ -25,7 +25,6 @@ from homeassistant.const import (
|
||||||
)
|
)
|
||||||
from homeassistant.core import HomeAssistant, callback
|
from homeassistant.core import HomeAssistant, callback
|
||||||
from homeassistant.helpers.entity_platform import AddEntitiesCallback
|
from homeassistant.helpers.entity_platform import AddEntitiesCallback
|
||||||
from homeassistant.helpers.typing import StateType
|
|
||||||
from homeassistant.helpers.update_coordinator import CoordinatorEntity
|
from homeassistant.helpers.update_coordinator import CoordinatorEntity
|
||||||
|
|
||||||
from . import AccuWeatherDataUpdateCoordinator
|
from . import AccuWeatherDataUpdateCoordinator
|
||||||
|
@ -50,7 +49,7 @@ PARALLEL_UPDATES = 1
|
||||||
class AccuWeatherSensorDescriptionMixin:
|
class AccuWeatherSensorDescriptionMixin:
|
||||||
"""Mixin for AccuWeather sensor."""
|
"""Mixin for AccuWeather sensor."""
|
||||||
|
|
||||||
value_fn: Callable[[dict[str, Any]], StateType]
|
value_fn: Callable[[dict[str, Any]], str | int | float | None]
|
||||||
|
|
||||||
|
|
||||||
@dataclass
|
@dataclass
|
||||||
|
@ -59,7 +58,7 @@ class AccuWeatherSensorDescription(
|
||||||
):
|
):
|
||||||
"""Class describing AccuWeather sensor entities."""
|
"""Class describing AccuWeather sensor entities."""
|
||||||
|
|
||||||
attr_fn: Callable[[dict[str, Any]], dict[str, StateType]] = lambda _: {}
|
attr_fn: Callable[[dict[str, Any]], dict[str, Any]] = lambda _: {}
|
||||||
|
|
||||||
|
|
||||||
FORECAST_SENSOR_TYPES: tuple[AccuWeatherSensorDescription, ...] = (
|
FORECAST_SENSOR_TYPES: tuple[AccuWeatherSensorDescription, ...] = (
|
||||||
|
@ -428,7 +427,7 @@ class AccuWeatherSensor(
|
||||||
self.forecast_day = forecast_day
|
self.forecast_day = forecast_day
|
||||||
|
|
||||||
@property
|
@property
|
||||||
def native_value(self) -> StateType:
|
def native_value(self) -> str | int | float | None:
|
||||||
"""Return the state."""
|
"""Return the state."""
|
||||||
return self.entity_description.value_fn(self._sensor_data)
|
return self.entity_description.value_fn(self._sensor_data)
|
||||||
|
|
||||||
|
|
Loading…
Add table
Add a link
Reference in a new issue