diff --git a/src/pydase/task/decorator.py b/src/pydase/task/decorator.py index ee704a9..224e119 100644 --- a/src/pydase/task/decorator.py +++ b/src/pydase/task/decorator.py @@ -1,4 +1,5 @@ import asyncio +import functools import logging from collections.abc import Callable, Coroutine from typing import Any, Concatenate, ParamSpec, TypeVar @@ -17,6 +18,7 @@ def task( def decorator( func: Callable[Concatenate[Any, P], Coroutine[None, None, R]], ) -> Task[P, R]: + @functools.wraps(func) async def wrapper(self: Any, *args: P.args, **kwargs: P.kwargs) -> R | None: try: return await func(self, *args, **kwargs) diff --git a/src/pydase/task/task.py b/src/pydase/task/task.py index 7b9a628..781988a 100644 --- a/src/pydase/task/task.py +++ b/src/pydase/task/task.py @@ -61,7 +61,7 @@ class Task(pydase.DataService, Generic[P, R]): self._result = task.result() - logger.info("Starting task") + logger.info("Starting task %s", self._func.__name__) if inspect.iscoroutinefunction(self._bound_func): res: Coroutine[None, None, R] = self._bound_func(*args, **kwargs) self._task = asyncio.create_task(res)