Improve opower generic typing (#95758)

pull/95784/head
Marc Mueller 2023-07-03 17:43:52 +02:00 committed by GitHub
parent 430a1bcb3d
commit 0a165bb35a
No known key found for this signature in database
GPG Key ID: 4AEE18F83AFDEB23
2 changed files with 2 additions and 2 deletions

View File

@ -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__(

View File

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