From 0a165bb35acbacb9eb5a57e67dfad68b02ad6e1f Mon Sep 17 00:00:00 2001 From: Marc Mueller <30130371+cdce8p@users.noreply.github.com> Date: Mon, 3 Jul 2023 17:43:52 +0200 Subject: [PATCH] Improve opower generic typing (#95758) --- homeassistant/components/opower/coordinator.py | 2 +- homeassistant/components/opower/sensor.py | 2 +- 2 files changed, 2 insertions(+), 2 deletions(-) diff --git a/homeassistant/components/opower/coordinator.py b/homeassistant/components/opower/coordinator.py index 4d40bb3356b..c331f45bc49 100644 --- a/homeassistant/components/opower/coordinator.py +++ b/homeassistant/components/opower/coordinator.py @@ -32,7 +32,7 @@ from .const import CONF_UTILITY, DOMAIN _LOGGER = logging.getLogger(__name__) -class OpowerCoordinator(DataUpdateCoordinator): +class OpowerCoordinator(DataUpdateCoordinator[dict[str, Forecast]]): """Handle fetching Opower data, updating sensors and inserting statistics.""" def __init__( diff --git a/homeassistant/components/opower/sensor.py b/homeassistant/components/opower/sensor.py index ef8d8eb884f..36f88a36e8a 100644 --- a/homeassistant/components/opower/sensor.py +++ b/homeassistant/components/opower/sensor.py @@ -170,7 +170,7 @@ async def async_setup_entry( coordinator: OpowerCoordinator = hass.data[DOMAIN][entry.entry_id] entities: list[OpowerSensor] = [] - forecasts: list[Forecast] = coordinator.data.values() + forecasts = coordinator.data.values() for forecast in forecasts: device_id = f"{coordinator.api.utility.subdomain()}_{forecast.account.utility_account_id}" device = DeviceInfo(