257 files

custom_components/maintenance_supporter/entity/triggers/state_change.py

@@ -17,7 +17,7 @@ from ...const import UNAVAILABLE_STATES
 from ...helpers.managed_timer import ManagedTimer
 
 if TYPE_CHECKING:
-    from ...sensor import MaintenanceSensor
+    from .base_trigger import TriggerHost
 
 from .base_trigger import BaseTrigger
 
@@ -48,7 +48,7 @@ class StateChangeTrigger(BaseTrigger):
     def __init__(
         self,
         hass: HomeAssistant,
-        entity: MaintenanceSensor,
+        entity: TriggerHost,
         trigger_config: dict[str, Any],
     ) -> None:
         """Initialize state change trigger."""