Switch gios to async_on_remove (#34499)

2020-04-21 14:16:37 +02:00
parent 6a4c25b668
commit 2584a07452
1 changed files with 3 additions and 5 deletions
--- 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.async_on_remove(
            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)
+        )

    async def async_update(self):
        """Update GIOS entity."""