257 files

custom_components/maintenance_supporter/entity/triggers/counter.py

@@ -8,7 +8,7 @@ from typing import TYPE_CHECKING, Any
 from homeassistant.core import HomeAssistant
 
 if TYPE_CHECKING:
-    from ...sensor import MaintenanceSensor
+    from .base_trigger import TriggerHost
 
 from ...helpers.trigger_fallback import counter_baseline
 from .base_trigger import BaseTrigger
@@ -27,7 +27,7 @@ class CounterTrigger(BaseTrigger):
     def __init__(
         self,
         hass: HomeAssistant,
-        entity: MaintenanceSensor,
+        entity: TriggerHost,
         trigger_config: dict[str, Any],
     ) -> None:
         """Initialize counter trigger."""