style fixes

Author: vgvoleg

tests/aio/query/conftest.py

@@ -1,6 +1,7 @@
 import pytest
 from ydb.aio.query.session import QuerySessionAsync
 
+
 @pytest.fixture
 async def session(driver):
     session = QuerySessionAsync(driver)

ydb/aio/query/base.py

@@ -1,9 +1,10 @@
 from .. import _utilities
 
+
 class AsyncResponseContextIterator(_utilities.AsyncResponseIterator):
     async def __aenter__(self) -> "AsyncResponseContextIterator":
         return self
 
     async def __aexit__(self, exc_type, exc_val, exc_tb):
         async for _ in self:
-            pass
+            pass

ydb/aio/query/session.py

@@ -17,6 +17,7 @@
     QuerySessionStateEnum,
 )
 
+
 class QuerySessionAsync(BaseQuerySession):
     """Session object for Query Service. It is not recommended to control
     session's lifecycle manually - use a QuerySessionPool is always a better choise.
@@ -45,7 +46,6 @@ async def _attach(self) -> None:
 
         self._loop.create_task(self._check_session_status_loop(), name="check session status task")
 
-
     async def _check_session_status_loop(self) -> None:
         try:
             async for status in self._status_stream:
@@ -85,7 +85,7 @@ async def create(self) -> "QuerySessionAsync":
 
         return self
 
-    def transaction(self, tx_mode = None) -> base.IQueryTxContext:
+    def transaction(self, tx_mode=None) -> base.IQueryTxContext:
         self._state._check_session_ready_to_use()
         tx_mode = tx_mode if tx_mode else _ydb_query_public.QuerySerializableReadWrite()