more style fixes

Author: vgvoleg

ydb/_grpc/grpcwrapper/ydb_query.py

@@ -147,8 +147,14 @@ class TransactionControl(IToProto):
 
     def to_proto(self) -> ydb_query_pb2.TransactionControl:
         if self.tx_id:
-            return ydb_query_pb2.TransactionControl(tx_id=self.tx_id,commit_tx=self.commit_tx)
-        return ydb_query_pb2.TransactionControl(begin_tx=self.begin_tx.to_proto(), commit_tx=self.commit_tx)
+            return ydb_query_pb2.TransactionControl(
+                tx_id=self.tx_id,
+                commit_tx=self.commit_tx,
+            )
+        return ydb_query_pb2.TransactionControl(
+            begin_tx=self.begin_tx.to_proto(),
+            commit_tx=self.commit_tx,
+        )
 
 
 @dataclass

ydb/query/base.py

@@ -83,7 +83,7 @@ def __init__(
         driver: SupportedDriverType,
         session_state: IQuerySessionState,
         session: IQuerySession,
-        tx_mode: BaseQueryTxMode = None
+        tx_mode: BaseQueryTxMode = None,
     ):
         pass
 
@@ -132,7 +132,7 @@ def session(self) -> IQuerySession:
 
 
 def create_execute_query_request(
-        query: str, session_id: str, tx_id: str = None, commit_tx: bool = False, tx_mode: BaseQueryTxMode = None
+    query: str, session_id: str, tx_id: str = None, commit_tx: bool = False, tx_mode: BaseQueryTxMode = None
 ):
     if tx_id:
         req = ydb_query.ExecuteQueryRequest(

ydb/query/session.py

@@ -28,7 +28,7 @@ class QuerySessionStateHelper(abc.ABC):
     _VALID_TRANSITIONS = {
         QuerySessionStateEnum.NOT_INITIALIZED: [QuerySessionStateEnum.CREATED],
         QuerySessionStateEnum.CREATED: [QuerySessionStateEnum.CLOSED],
-        QuerySessionStateEnum.CLOSED: []
+        QuerySessionStateEnum.CLOSED: [],
     }
 
     _READY_TO_USE = [