retry: fix decorator0.12

The previous one was buggy. We also must pass None to stop_max_delay and
not a boolean, otherwise it fails.

Change-Id: Ia0babdfbae3b6ce88ae37f79b39fc4d215801b95

1 files changed, 5 insertions, 5 deletions

diff --git a/tooz/drivers/_retry.py b/tooz/drivers/_retry.py
index 7fdd1d1..471bba1 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