diff --git a/hahomematic/platforms/entity.py b/hahomematic/platforms/entity.py index af7fbe25..69583081 100644 --- a/hahomematic/platforms/entity.py +++ b/hahomematic/platforms/entity.py @@ -846,7 +846,7 @@ async def wrapper(*args: Any, **kwargs: Any) -> Any: ) return return_value - setattr(func, "__service", True) + setattr(func, "ha_service", True) return wrapper # type: ignore[return-value] return decorator @@ -874,7 +874,7 @@ async def wrapper(*args: P.args, **kwargs: P.kwargs) -> T: ) raise - setattr(func, "__service", True) + setattr(wrapper, "ha_service", True) return wrapper return decorator @@ -887,5 +887,5 @@ def get_service_calls(obj: object) -> dict[str, Callable]: for name in dir(obj) if not name.startswith("_") and callable(getattr(obj, name)) - and hasattr(getattr(obj, name), "__service") + and hasattr(getattr(obj, name), "ha_service") }