diff --git a/retrying.py b/retrying.py index 9f0a2a0..7a00a8d 100644 --- a/retrying.py +++ b/retrying.py @@ -55,6 +55,8 @@ def retry(*dargs, **dkw): # support both @retry and @retry() as valid syntax if len(dargs) == 1 and callable(dargs[0]): def wrap_simple(f): + + @six.wraps(f) def wrapped_f(*args, **kw): return Retrying().call(f, *args, **kw) @@ -64,6 +66,8 @@ def retry(*dargs, **dkw): else: def wrap(f): + + @six.wraps(f) def wrapped_f(*args, **kw): return Retrying(*dargs, **dkw).call(f, *args, **kw)