Skip to content

Commit d2b58f1

Browse files
fix: annotate async status cache
Signed-off-by: Aakash Saravanan <aakash.saravanan@databricks.com>
1 parent 2736b2d commit d2b58f1

1 file changed

Lines changed: 3 additions & 1 deletion

File tree

‎src/databricks/sql/backend/thrift_backend.py‎

Lines changed: 3 additions & 1 deletion
Original file line numberDiff line numberDiff line change
@@ -268,7 +268,9 @@ def __init__(
268268
raise
269269

270270
self._request_lock = threading.RLock()
271-
self._terminal_operation_statuses = weakref.WeakKeyDictionary()
271+
self._terminal_operation_statuses: weakref.WeakKeyDictionary[
272+
CommandId, ttypes.TGetOperationStatusResp
273+
] = weakref.WeakKeyDictionary()
272274
self._session_id_hex = None
273275

274276
@property

0 commit comments

Comments
 (0)