diff --git a/homeassistant/components/shelly/device_trigger.py b/homeassistant/components/shelly/device_trigger.py index ae2eb7d6440..1231f670d49 100644 --- a/homeassistant/components/shelly/device_trigger.py +++ b/homeassistant/components/shelly/device_trigger.py @@ -1,7 +1,7 @@ """Provides device triggers for Shelly.""" from __future__ import annotations -from typing import Any, Final +from typing import Final import voluptuous as vol @@ -9,7 +9,10 @@ from homeassistant.components.automation import ( AutomationActionType, AutomationTriggerInfo, ) -from homeassistant.components.device_automation import DEVICE_TRIGGER_BASE_SCHEMA +from homeassistant.components.device_automation import ( + DEVICE_TRIGGER_BASE_SCHEMA, + GetAutomationsResult, +) from homeassistant.components.device_automation.exceptions import ( InvalidDeviceAutomationConfig, ) @@ -72,8 +75,8 @@ def append_input_triggers( async def async_validate_trigger_config( - hass: HomeAssistant, config: dict[str, Any] -) -> dict[str, Any]: + hass: HomeAssistant, config: ConfigType +) -> ConfigType: """Validate config.""" config = TRIGGER_SCHEMA(config) @@ -108,9 +111,9 @@ async def async_validate_trigger_config( async def async_get_triggers( hass: HomeAssistant, device_id: str -) -> list[dict[str, str]]: +) -> GetAutomationsResult: """List device triggers for Shelly devices.""" - triggers: list[dict[str, str]] = [] + triggers: GetAutomationsResult = [] if rpc_wrapper := get_rpc_device_wrapper(hass, device_id): input_triggers = get_rpc_input_triggers(rpc_wrapper.device)