Change a bit import.

ionelmc · Feb 24, 2015 · 4cf041c · 4cf041c
1 parent f315706
commit 4cf041c
Showing 1 changed file with 2 additions and 2 deletions.
diff --git a/src/aspectlib/contrib.py b/src/aspectlib/contrib.py
@@ -1,12 +1,12 @@
 from logging import getLogger
-from time import sleep
+import time
 
 from aspectlib import Aspect
 
 logger = getLogger(__name__)
 
 
-def retry(func=None, retries=5, backoff=None, exceptions=(IOError, OSError, EOFError), cleanup=None, sleep=sleep):
+def retry(func=None, retries=5, backoff=None, exceptions=(IOError, OSError, EOFError), cleanup=None, sleep=time.sleep):
     """
     Decorator that retries the call ``retries`` times if ``func`` raises ``exceptions``. Can use a ``backoff`` function
     to sleep till next retry.