diff --git a/kasa/interfaces/fan.py b/kasa/interfaces/fan.py index ade00928..9462ad88 100644 --- a/kasa/interfaces/fan.py +++ b/kasa/interfaces/fan.py @@ -3,8 +3,9 @@ from __future__ import annotations from abc import ABC, abstractmethod +from typing import Annotated -from ..module import Module +from ..module import FeatureAttribute, Module class Fan(Module, ABC): @@ -12,9 +13,11 @@ class Fan(Module, ABC): @property @abstractmethod - def fan_speed_level(self) -> int: + def fan_speed_level(self) -> Annotated[int, FeatureAttribute()]: """Return fan speed level.""" @abstractmethod - async def set_fan_speed_level(self, level: int) -> dict: + async def set_fan_speed_level( + self, level: int + ) -> Annotated[dict, FeatureAttribute()]: """Set fan speed level."""