88from rachiopy import Rachio
99from requests .exceptions import Timeout
1010
11+ from homeassistant .config_entries import ConfigEntry
1112from homeassistant .core import HomeAssistant
1213from homeassistant .helpers .debounce import Debouncer
1314from homeassistant .helpers .update_coordinator import DataUpdateCoordinator , UpdateFailed
@@ -38,6 +39,7 @@ def __init__(
3839 self ,
3940 hass : HomeAssistant ,
4041 rachio : Rachio ,
42+ config_entry : ConfigEntry ,
4143 base_station ,
4244 base_count : int ,
4345 ) -> None :
@@ -48,6 +50,7 @@ def __init__(
4850 super ().__init__ (
4951 hass ,
5052 _LOGGER ,
53+ config_entry = config_entry ,
5154 name = f"{ DOMAIN } update coordinator" ,
5255 # To avoid exceeding the rate limit, increase polling interval for
5356 # each additional base station on the account
@@ -76,6 +79,7 @@ def __init__(
7679 self ,
7780 hass : HomeAssistant ,
7881 rachio : Rachio ,
82+ config_entry : ConfigEntry ,
7983 base_station ,
8084 ) -> None :
8185 """Initialize a Rachio schedule coordinator."""
@@ -85,6 +89,7 @@ def __init__(
8589 super ().__init__ (
8690 hass ,
8791 _LOGGER ,
92+ config_entry = config_entry ,
8893 name = f"{ DOMAIN } schedule update coordinator" ,
8994 update_interval = timedelta (minutes = 30 ),
9095 )
0 commit comments