From 2584a07452d21583814bd59d9e01a91e784f0f82 Mon Sep 17 00:00:00 2001 From: Maciej Bieniek Date: Tue, 21 Apr 2020 14:16:37 +0200 Subject: [PATCH] Switch gios to async_on_remove (#34499) --- homeassistant/components/gios/air_quality.py | 8 +++----- 1 file changed, 3 insertions(+), 5 deletions(-) diff --git a/homeassistant/components/gios/air_quality.py b/homeassistant/components/gios/air_quality.py index c8cd8be11c7..d180b0a0ddf 100644 --- a/homeassistant/components/gios/air_quality.py +++ b/homeassistant/components/gios/air_quality.py @@ -142,11 +142,9 @@ class GiosAirQuality(AirQualityEntity): async def async_added_to_hass(self): """Connect to dispatcher listening for entity data notifications.""" - self.coordinator.async_add_listener(self.async_write_ha_state) - - async def async_will_remove_from_hass(self): - """Disconnect from update signal.""" - self.coordinator.async_remove_listener(self.async_write_ha_state) + self.async_on_remove( + self.coordinator.async_add_listener(self.async_write_ha_state) + ) async def async_update(self): """Update GIOS entity."""