From 5edf2b3db33a42b646fe923580debff6b58a1122 Mon Sep 17 00:00:00 2001
From: Julien Danjou <julien@danjou.info>
Date: Mon, 19 Jan 2015 18:11:13 +0100
Subject: [PATCH] retry: fix decorator

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

Change-Id: Ia0babdfbae3b6ce88ae37f79b39fc4d215801b95
---
 tooz/drivers/_retry.py | 10 +++++-----
 1 file changed, 5 insertions(+), 5 deletions(-)

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