fixup: lease typed dict

python-zk · Nov 17, 2022 · 16f1829 · 16f1829
1 parent 43466c7
commit 16f1829
Show file tree

Hide file tree

Showing 2 changed files with 9 additions and 5 deletions.
diff --git a/kazoo/recipe/lease.py b/kazoo/recipe/lease.py
@@ -14,15 +14,16 @@
 
 from kazoo.exceptions import CancelledError
 
+from typing_extensions import TypedDict
+
 if TYPE_CHECKING:
     from kazoo.client import KazooClient
     from typing import Callable, Optional
-    from typing_extensions import TypedDict
 
-    class LeaseData(TypedDict):
-        version: int
-        holder: str
-        end: str
+class LeaseData(TypedDict):
+    version: int
+    holder: str
+    end: str
 
 
 class NonBlockingLease(object):

diff --git a/setup.cfg b/setup.cfg
@@ -42,6 +42,7 @@ packages = find:
 install_requires =
     six
     selectors2>=2.0.2 ; python_version < "3.4.0"
+    typing-extensions
 
 [aliases]
 release = sdist bdist_wheel
@@ -66,6 +67,7 @@ test =
 
 typing =
     mypy
+    types-six
     types-mock
 
 eventlet =
@@ -86,5 +88,6 @@ alldeps =
     %(gevent)s
     %(sasl)s
     %(docs)s
+    %(typing)s