Update __init__.py

rospogrigio · Nov 16, 2023 · 7e3284d · 7e3284d
1 parent 914381b
commit 7e3284d
Showing 1 changed file with 10 additions and 6 deletions.
diff --git a/custom_components/airbnk_mqtt/__init__.py b/custom_components/airbnk_mqtt/__init__.py
@@ -146,12 +146,16 @@ async def async_options_updated(hass, entry):
 async def async_unload_entry(hass, config_entry):
     """Unload a config entry."""
     _LOGGER.debug("Unloading %s %s", config_entry.entry_id, config_entry.data)
-    await asyncio.wait(asyncio.create_task(
-        [
-            hass.config_entries.async_forward_entry_unload(config_entry, component)
-            for component in COMPONENT_TYPES
-        ]
-    ))
+
+    # Create a list of coroutines to be awaited
+    tasks = [
+        hass.config_entries.async_forward_entry_unload(config_entry, component)
+        for component in COMPONENT_TYPES
+    ]
+
+    # Await each coroutine individually
+    await asyncio.gather(*tasks)
+
     for dev_id, device in hass.data[DOMAIN][AIRBNK_DEVICES].items():
         _LOGGER.debug("Unsubscribing %s", dev_id)
         await device.mqtt_unsubscribe()