diff --git a/homeassistant/components/drop_connect/switch.py b/homeassistant/components/drop_connect/switch.py index 1cd7fbf39f4..b0ebe4b5a85 100644 --- a/homeassistant/components/drop_connect/switch.py +++ b/homeassistant/components/drop_connect/switch.py @@ -2,7 +2,7 @@ from __future__ import annotations -from collections.abc import Callable +from collections.abc import Awaitable, Callable from dataclasses import dataclass import logging from typing import Any @@ -42,7 +42,7 @@ class DROPSwitchEntityDescription(SwitchEntityDescription): """Describes DROP switch entity.""" value_fn: Callable[[DROPDeviceDataUpdateCoordinator], int | None] - set_fn: Callable + set_fn: Callable[[DROPDeviceDataUpdateCoordinator, int], Awaitable[Any]] SWITCHES: list[DROPSwitchEntityDescription] = [