Fix ConfigEntry in coordinator

alexdelprete · Jul 6, 2024 · 400c29d · 400c29d
1 parent ffd115b
commit 400c29d
Showing 1 changed file with 3 additions and 5 deletions.
diff --git a/custom_components/sinapsi_alfa/coordinator.py b/custom_components/sinapsi_alfa/coordinator.py
@@ -6,10 +6,10 @@
 import logging
 from datetime import datetime, timedelta
 
+from homeassistant.config_entries import ConfigEntry
 from homeassistant.core import HomeAssistant
 from homeassistant.helpers.update_coordinator import DataUpdateCoordinator, UpdateFailed
 
-from . import SinapsiAlfaConfigEntry
 from .api import SinapsiAlfaAPI
 from .const import (
     CONF_HOST,
@@ -27,11 +27,9 @@
 class SinapsiAlfaCoordinator(DataUpdateCoordinator):
     """Class to manage fetching data from the API."""
 
-    config_entry: SinapsiAlfaConfigEntry
+    config_entry: ConfigEntry
 
-    def __init__(
-        self, hass: HomeAssistant, config_entry: SinapsiAlfaConfigEntry
-    ) -> None:
+    def __init__(self, hass: HomeAssistant, config_entry: ConfigEntry) -> None:
         """Initialize data update coordinator."""
 
         # get scan_interval from user config