typing

pomponchik · Mar 31, 2024 · 7f0c6b0 · 7f0c6b0
1 parent 8a2f150
commit 7f0c6b0
Showing 1 changed file with 2 additions and 2 deletions.
diff --git a/locklib/locks/smart_lock/lock.py b/locklib/locks/smart_lock/lock.py
@@ -4,7 +4,7 @@
  from threading import Lock, get_ident as get_native_id # get_native_id is available only since python 3.8
 
 from collections import deque
-from typing import Type, Deque, Dict
+from typing import Type, Deque, Dict, Optional
 from types import TracebackType
 
 from locklib.locks.smart_lock.graph import LocksGraph
@@ -61,5 +61,5 @@ def release(self) -> None:
  def __enter__(self) -> None:
  self.acquire()
 
- def __exit__(self, exception_type: Type[BaseException], exception_value: BaseException, traceback: TracebackType) -> None:
+ def __exit__(self, exception_type: Optional[Type[BaseException]], exception_value: Optional[BaseException], traceback: Optional[TracebackType]) -> None:
  self.release()