use functools.wraps in task decorator

nipype · Apr 6, 2022 · 418eca7 · 418eca7
1 parent 92bc8c3
commit 418eca7
Showing 1 changed file with 2 additions and 0 deletions.
diff --git a/pydra/mark/functions.py b/pydra/mark/functions.py
@@ -1,4 +1,5 @@
 """ Decorators to apply to functions used in Pydra workflows """
+from functools import wraps
 
 
 def annotate(annotation):
@@ -40,6 +41,7 @@ def task(func):
     """
     from ..engine.task import FunctionTask
 
+    @wraps(func)
     def decorate(**kwargs):
         return FunctionTask(func=func, **kwargs)