diff options
Diffstat (limited to 'timeout_decorator/timeout_decorator.py')
-rw-r--r-- | timeout_decorator/timeout_decorator.py | 5 |
1 files changed, 4 insertions, 1 deletions
diff --git a/timeout_decorator/timeout_decorator.py b/timeout_decorator/timeout_decorator.py index 466f33d..aadc586 100644 --- a/timeout_decorator/timeout_decorator.py +++ b/timeout_decorator/timeout_decorator.py @@ -73,7 +73,10 @@ def timeout(seconds=None, use_signals=True, timeout_exception=TimeoutError): signal.signal(signal.SIGALRM, old) return new_function else: - return _Timeout(function, timeout_exception, seconds) + @wraps(function) + def new_function(*args, **kwargs): + return _Timeout(function, timeout_exception, seconds) + return new_function return decorate |