257 files

custom_components/maintenance_supporter/entity/triggers/threshold.py

@@ -10,7 +10,7 @@ from homeassistant.core import HomeAssistant, callback
 from homeassistant.util import dt as dt_util
 
 if TYPE_CHECKING:
-    from ...sensor import MaintenanceSensor
+    from .base_trigger import TriggerHost
 
 from ...helpers.managed_timer import ManagedTimer
 from ...helpers.trigger_fallback import threshold_exceeds
@@ -32,7 +32,7 @@ class ThresholdTrigger(BaseTrigger):
     def __init__(
         self,
         hass: HomeAssistant,
-        entity: MaintenanceSensor,
+        entity: TriggerHost,
         trigger_config: dict[str, Any],
     ) -> None:
         """Initialize threshold trigger."""