fixes

Author: vgvoleg

ydb/query/session.py

@@ -105,7 +105,7 @@ def wrapper_create_session(
     rpc_state: RpcState,
     response_pb: _apis.ydb_query.CreateSessionResponse,
     session_state: QuerySessionState,
-    session: "BaseQuerySession"
+    session: "BaseQuerySession",
 ) -> "BaseQuerySession":
     message = _ydb_query.CreateSessionResponse.from_proto(response_pb)
     issues._process_response(message.status)
@@ -117,7 +117,7 @@ def wrapper_delete_session(
     rpc_state: RpcState,
     response_pb: _apis.ydb_query.DeleteSessionResponse,
     session_state: QuerySessionState,
-    session: "BaseQuerySession"
+    session: "BaseQuerySession",
 ) -> "BaseQuerySession":
     message = _ydb_query.DeleteSessionResponse.from_proto(response_pb)
     issues._process_response(message.status)

ydb/query/transaction.py

@@ -52,7 +52,9 @@ def terminal(cls, state: QueryTxStateEnum) -> bool:
 
 def reset_tx_id_handler(func):
     @functools.wraps(func)
-    def decorator(rpc_state, response_pb, session_state: base.IQuerySessionState, tx_state: QueryTxState, *args, **kwargs):
+    def decorator(
+        rpc_state, response_pb, session_state: base.IQuerySessionState, tx_state: QueryTxState, *args, **kwargs
+    ):
         try:
             return func(rpc_state, response_pb, session_state, tx_state, *args, **kwargs)
         except issues.Error: