@@ -40,17 +40,24 @@ class AwairResult:
4040class AwairDataUpdateCoordinator (DataUpdateCoordinator [dict [str , AwairResult ]]):
4141 """Define a wrapper class to update Awair data."""
4242
43+ config_entry : AwairConfigEntry
44+
4345 def __init__ (
4446 self ,
4547 hass : HomeAssistant ,
46- config_entry : ConfigEntry ,
48+ config_entry : AwairConfigEntry ,
4749 update_interval : timedelta | None ,
4850 ) -> None :
4951 """Set up the AwairDataUpdateCoordinator class."""
50- self ._config_entry = config_entry
5152 self .title = config_entry .title
5253
53- super ().__init__ (hass , LOGGER , name = DOMAIN , update_interval = update_interval )
54+ super ().__init__ (
55+ hass ,
56+ LOGGER ,
57+ config_entry = config_entry ,
58+ name = DOMAIN ,
59+ update_interval = update_interval ,
60+ )
5461
5562 async def _fetch_air_data (self , device : AwairBaseDevice ) -> AwairResult :
5663 """Fetch latest air quality data."""
@@ -64,7 +71,10 @@ class AwairCloudDataUpdateCoordinator(AwairDataUpdateCoordinator):
6471 """Define a wrapper class to update Awair data from Cloud API."""
6572
6673 def __init__ (
67- self , hass : HomeAssistant , config_entry : ConfigEntry , session : ClientSession
74+ self ,
75+ hass : HomeAssistant ,
76+ config_entry : AwairConfigEntry ,
77+ session : ClientSession ,
6878 ) -> None :
6979 """Set up the AwairCloudDataUpdateCoordinator class."""
7080 access_token = config_entry .data [CONF_ACCESS_TOKEN ]
@@ -95,7 +105,10 @@ class AwairLocalDataUpdateCoordinator(AwairDataUpdateCoordinator):
95105 _device : AwairLocalDevice | None = None
96106
97107 def __init__ (
98- self , hass : HomeAssistant , config_entry : ConfigEntry , session : ClientSession
108+ self ,
109+ hass : HomeAssistant ,
110+ config_entry : AwairConfigEntry ,
111+ session : ClientSession ,
99112 ) -> None :
100113 """Set up the AwairLocalDataUpdateCoordinator class."""
101114 self ._awair = AwairLocal (
0 commit comments