From 06a752aa927246fb347980b82f0c6bdc59ae6640 Mon Sep 17 00:00:00 2001 From: "J. Nick Koston" Date: Tue, 2 Apr 2024 16:09:27 -1000 Subject: [PATCH] Small speed ups to ambient_station (#114698) --- homeassistant/components/ambient_station/__init__.py | 5 +++-- 1 file changed, 3 insertions(+), 2 deletions(-) diff --git a/homeassistant/components/ambient_station/__init__.py b/homeassistant/components/ambient_station/__init__.py index 0984e21a722..b55a7b866cc 100644 --- a/homeassistant/components/ambient_station/__init__.py +++ b/homeassistant/components/ambient_station/__init__.py @@ -93,7 +93,7 @@ async def async_unload_entry(hass: HomeAssistant, entry: ConfigEntry) -> bool: unload_ok = await hass.config_entries.async_unload_platforms(entry, PLATFORMS) if unload_ok: ambient = hass.data[DOMAIN].pop(entry.entry_id) - hass.async_create_task(ambient.ws_disconnect()) + hass.async_create_task(ambient.ws_disconnect(), eager_start=True) return unload_ok @@ -179,7 +179,8 @@ class AmbientStation: self._hass.async_create_task( self._hass.config_entries.async_forward_entry_setups( self._entry, PLATFORMS - ) + ), + eager_start=True, ) self._entry_setup_complete = True self._ws_reconnect_delay = DEFAULT_SOCKET_MIN_RETRY