Remove the use of StateType from AccuWeather (#97109)

This commit is contained in:
Franck Nijhof 2023-07-23 23:30:37 +02:00 committed by GitHub
parent 69d7b035e0
commit 5e88ca23b3
No known key found for this signature in database
GPG key ID: 4AEE18F83AFDEB23

View file

@ -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)