From 2eb2bdd61558760439240205f448b6eb7befa252 Mon Sep 17 00:00:00 2001
From: Nicholas Romyn <13968908+nromyn@users.noreply.github.com>
Date: Wed, 6 Nov 2024 02:25:18 -0500
Subject: [PATCH] Consolidating async_add_entities into one call in Ecobee
 (#129917)

* Consolidating async_add_entities into one call.

* changing to comprehension.
---
 homeassistant/components/ecobee/switch.py | 33 ++++++++++++-----------
 1 file changed, 17 insertions(+), 16 deletions(-)

diff --git a/homeassistant/components/ecobee/switch.py b/homeassistant/components/ecobee/switch.py
index 67be78fb21d..89ee433c072 100644
--- a/homeassistant/components/ecobee/switch.py
+++ b/homeassistant/components/ecobee/switch.py
@@ -31,25 +31,26 @@ async def async_setup_entry(
     """Set up the ecobee thermostat switch entity."""
     data: EcobeeData = hass.data[DOMAIN]
 
-    async_add_entities(
-        [
-            EcobeeVentilator20MinSwitch(
-                data,
-                index,
-                (await dt_util.async_get_time_zone(thermostat["location"]["timeZone"]))
-                or dt_util.get_default_time_zone(),
-            )
+    entities: list[SwitchEntity] = [
+        EcobeeVentilator20MinSwitch(
+            data,
+            index,
+            (await dt_util.async_get_time_zone(thermostat["location"]["timeZone"]))
+            or dt_util.get_default_time_zone(),
+        )
+        for index, thermostat in enumerate(data.ecobee.thermostats)
+        if thermostat["settings"]["ventilatorType"] != "none"
+    ]
+
+    entities.extend(
+        (
+            EcobeeSwitchAuxHeatOnly(data, index)
             for index, thermostat in enumerate(data.ecobee.thermostats)
-            if thermostat["settings"]["ventilatorType"] != "none"
-        ],
-        update_before_add=True,
+            if thermostat["settings"]["hasHeatPump"]
+        )
     )
 
-    async_add_entities(
-        EcobeeSwitchAuxHeatOnly(data, index)
-        for index, thermostat in enumerate(data.ecobee.thermostats)
-        if thermostat["settings"]["hasHeatPump"]
-    )
+    async_add_entities(entities, update_before_add=True)
 
 
 class EcobeeVentilator20MinSwitch(EcobeeBaseEntity, SwitchEntity):