diff options
-rw-r--r-- | timeout_decorator/timeout_decorator.py | 3 |
1 files changed, 2 insertions, 1 deletions
diff --git a/timeout_decorator/timeout_decorator.py b/timeout_decorator/timeout_decorator.py index aadc586..58802a5 100644 --- a/timeout_decorator/timeout_decorator.py +++ b/timeout_decorator/timeout_decorator.py @@ -75,7 +75,8 @@ def timeout(seconds=None, use_signals=True, timeout_exception=TimeoutError): else: @wraps(function) def new_function(*args, **kwargs): - return _Timeout(function, timeout_exception, seconds) + timeout_wrapper = _Timeout(function, timeout_exception, seconds) + return timeout_wrapper(*args, **kwargs) return new_function return decorate |