From 89ec04572676f33c59d349f6ef912db06a116ca3 Mon Sep 17 00:00:00 2001 From: Andrey Khrolenok Date: Fri, 3 May 2024 20:25:20 +0300 Subject: [PATCH] Fix sensor_state_listener --- custom_components/iaquk/__init__.py | 6 +++--- 1 file changed, 3 insertions(+), 3 deletions(-) diff --git a/custom_components/iaquk/__init__.py b/custom_components/iaquk/__init__.py index dcc2245..dc37665 100644 --- a/custom_components/iaquk/__init__.py +++ b/custom_components/iaquk/__init__.py @@ -22,7 +22,7 @@ UNIT_NOT_RECOGNIZED_TEMPLATE, UnitOfTemperature, ) -from homeassistant.core import State, callback +from homeassistant.core import Event, EventStateChangedData, State, callback from homeassistant.helpers import discovery import homeassistant.helpers.config_validation as cv from homeassistant.helpers.event import async_track_state_change_event @@ -177,7 +177,7 @@ def async_added_to_hass(self): # pylint: disable=unused-argument @callback - def sensor_state_listener(entity, old_state, new_state): + def sensor_state_listener(event: Event[EventStateChangedData]) -> None: """Handle device state changes.""" self.update() @@ -199,7 +199,7 @@ def sensor_startup(event): ) async_track_state_change_event(self.hass, entity_ids, sensor_state_listener) - sensor_state_listener(None, None, None) # Force first update + self.update() # Force first update if not self._added: self._added = True