diff --git a/homeassistant/components/tuya/tuya_device_handlers/builder/base_quirk.py b/homeassistant/components/tuya/tuya_device_handlers/builder/base_quirk.py index 4f197d6ce8bb..ef60e3aa0b58 100644 --- a/homeassistant/components/tuya/tuya_device_handlers/builder/base_quirk.py +++ b/homeassistant/components/tuya/tuya_device_handlers/builder/base_quirk.py @@ -8,7 +8,7 @@ import inspect import pathlib from typing import TYPE_CHECKING, Any, Self -from ..helpers import ( +from tuya_device_handlers.helpers import ( TuyaClimateHVACMode, TuyaCoverDeviceClass, TuyaDeviceCategory, @@ -18,10 +18,15 @@ from ..helpers import ( ) if TYPE_CHECKING: + from tuya_sharing import CustomerDevice + + from ..helpers import TuyaIntegerTypeDefinition from ..registry import QuirksRegistry -type TuyaIntegerConversionFunction = Callable[[Any, Any, Any], float] +type TuyaIntegerConversionFunction = Callable[ + [CustomerDevice, TuyaIntegerTypeDefinition, Any], Any +] """Start conversion function: Args: