summaryrefslogtreecommitdiff
diff options
context:
space:
mode:
-rwxr-xr-xzookeeper.py5
1 files changed, 4 insertions, 1 deletions
diff --git a/zookeeper.py b/zookeeper.py
index 42771c9..e2e6ce4 100755
--- a/zookeeper.py
+++ b/zookeeper.py
@@ -36,7 +36,10 @@ def obtain_lease(
"""Obtain the named lease before invoking a function and skip
invoking the function if the lease cannot be obtained.
- >>> @obtain_lease(lease_id='zookeeper_doctest')
+ >>> @obtain_lease(
+ ... lease_id='zookeeper_doctest',
+ ... initial_duration=datetime.timedelta(seconds=10),
+ ... )
... def f(name: str) -> int:
... print(f'Hello, {name}')
... return 123