Fix ConfigEntry in coordinator

alexdelprete · Jul 6, 2024 · 9713b96 · 9713b96
1 parent e59f98f
commit 9713b96
Showing 1 changed file with 3 additions and 5 deletions.
diff --git a/custom_components/abb_powerone_pvi_sunspec/coordinator.py b/custom_components/abb_powerone_pvi_sunspec/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 ABBPowerOneFimerConfigEntry
 from .api import ABBPowerOneFimerAPI
 from .const import (
     CONF_BASE_ADDR,
@@ -29,11 +29,9 @@
 class ABBPowerOneFimerCoordinator(DataUpdateCoordinator):
     """Class to manage fetching data from the API."""
 
-    config_entry: ABBPowerOneFimerConfigEntry
+    config_entry: ConfigEntry
 
-    def __init__(
-        self, hass: HomeAssistant, config_entry: ABBPowerOneFimerConfigEntry
-    ) -> None:
+    def __init__(self, hass: HomeAssistant, config_entry: ConfigEntry) -> None:
         """Initialize data update coordinator."""
 
         # get scan_interval from user config