From 52fd63acbce6d5b31f751cf82bbdca5ae39b8803 Mon Sep 17 00:00:00 2001 From: Robert Hillis Date: Wed, 10 Aug 2022 18:05:32 -0400 Subject: [PATCH] Use generators for async_add_entities in Accuweather (#76574) --- .../components/accuweather/sensor.py | 22 +++++++++---------- 1 file changed, 11 insertions(+), 11 deletions(-) diff --git a/homeassistant/components/accuweather/sensor.py b/homeassistant/components/accuweather/sensor.py index 72182d4d635..c13dedcdceb 100644 --- a/homeassistant/components/accuweather/sensor.py +++ b/homeassistant/components/accuweather/sensor.py @@ -320,19 +320,19 @@ async def async_setup_entry( coordinator: AccuWeatherDataUpdateCoordinator = hass.data[DOMAIN][entry.entry_id] - sensors: list[AccuWeatherSensor] = [] - for description in SENSOR_TYPES: - sensors.append(AccuWeatherSensor(coordinator, description)) + sensors = [ + AccuWeatherSensor(coordinator, description) for description in SENSOR_TYPES + ] if coordinator.forecast: - for description in FORECAST_SENSOR_TYPES: - for day in range(MAX_FORECAST_DAYS + 1): - # Some air quality/allergy sensors are only available for certain - # locations. - if description.key in coordinator.data[ATTR_FORECAST][0]: - sensors.append( - AccuWeatherSensor(coordinator, description, forecast_day=day) - ) + # Some air quality/allergy sensors are only available for certain + # locations. + sensors.extend( + AccuWeatherSensor(coordinator, description, forecast_day=day) + for description in FORECAST_SENSOR_TYPES + for day in range(MAX_FORECAST_DAYS + 1) + if description.key in coordinator.data[ATTR_FORECAST][0] + ) async_add_entities(sensors)