diff --git a/homeassistant/components/aurora/binary_sensor.py b/homeassistant/components/aurora/binary_sensor.py index d817ea51988..49e25e55950 100644 --- a/homeassistant/components/aurora/binary_sensor.py +++ b/homeassistant/components/aurora/binary_sensor.py @@ -1,4 +1,6 @@ """Support for Aurora Forecast binary sensor.""" +from __future__ import annotations + from homeassistant.components.binary_sensor import BinarySensorEntity from homeassistant.config_entries import ConfigEntry from homeassistant.core import HomeAssistant @@ -27,6 +29,6 @@ class AuroraSensor(AuroraEntity, BinarySensorEntity): """Implementation of an aurora sensor.""" @property - def is_on(self): + def is_on(self) -> bool: """Return true if aurora is visible.""" return self.coordinator.data > self.coordinator.threshold diff --git a/homeassistant/components/aurora/config_flow.py b/homeassistant/components/aurora/config_flow.py index 95e66ff226e..a1971884ead 100644 --- a/homeassistant/components/aurora/config_flow.py +++ b/homeassistant/components/aurora/config_flow.py @@ -51,7 +51,7 @@ class ConfigFlow(config_entries.ConfigFlow, domain=DOMAIN): self, user_input: dict[str, Any] | None = None ) -> FlowResult: """Handle the initial step.""" - errors = {} + errors: dict[str, str] = {} if user_input is not None: longitude = user_input[CONF_LONGITUDE] diff --git a/homeassistant/components/aurora/coordinator.py b/homeassistant/components/aurora/coordinator.py index 9d4eb0aa681..8195f6d30ec 100644 --- a/homeassistant/components/aurora/coordinator.py +++ b/homeassistant/components/aurora/coordinator.py @@ -1,4 +1,5 @@ """The aurora component.""" +from __future__ import annotations from datetime import timedelta import logging @@ -12,7 +13,7 @@ from homeassistant.helpers.update_coordinator import DataUpdateCoordinator, Upda _LOGGER = logging.getLogger(__name__) -class AuroraDataUpdateCoordinator(DataUpdateCoordinator): +class AuroraDataUpdateCoordinator(DataUpdateCoordinator[int]): """Class to manage fetching data from the NOAA Aurora API.""" def __init__( @@ -37,7 +38,7 @@ class AuroraDataUpdateCoordinator(DataUpdateCoordinator): self.longitude = int(longitude) self.threshold = int(threshold) - async def _async_update_data(self): + async def _async_update_data(self) -> int: """Fetch the data from the NOAA Aurora Forecast.""" try: diff --git a/homeassistant/components/aurora/sensor.py b/homeassistant/components/aurora/sensor.py index f44cc23f832..3ad36591f15 100644 --- a/homeassistant/components/aurora/sensor.py +++ b/homeassistant/components/aurora/sensor.py @@ -1,4 +1,6 @@ """Support for Aurora Forecast sensor.""" +from __future__ import annotations + from homeassistant.components.sensor import SensorEntity, SensorStateClass from homeassistant.config_entries import ConfigEntry from homeassistant.const import PERCENTAGE @@ -31,6 +33,6 @@ class AuroraSensor(AuroraEntity, SensorEntity): _attr_state_class = SensorStateClass.MEASUREMENT @property - def native_value(self): + def native_value(self) -> int: """Return % chance the aurora is visible.""" return self.coordinator.data