diff --git a/tooz/drivers/_retry.py b/tooz/drivers/_retry.py index 7fdd1d16..471bba17 100644 --- a/tooz/drivers/_retry.py +++ b/tooz/drivers/_retry.py @@ -32,11 +32,11 @@ RETRYING_KWARGS = dict( def retry(**kwargs): - def _retry(f): - k = RETRYING_KWARGS.copy() - k.update(kwargs) - return retrying.retry(**k)(f) - return _retry + delay = kwargs.get('stop_max_delay', None) + kwargs['stop_max_delay'] = delay if delay not in (True, False) else None + k = RETRYING_KWARGS.copy() + k.update(kwargs) + return retrying.retry(**k) Retrying = retrying.Retrying