style fixes

Author: vgvoleg

ydb/aio/query/pool.py

@@ -36,7 +36,9 @@ def checkout(self) -> "SimpleQuerySessionCheckoutAsync":
 
         return SimpleQuerySessionCheckoutAsync(self)
 
-    async def retry_operation_async(self, callee: Callable, retry_settings: Optional[RetrySettings] = None, *args, **kwargs):
+    async def retry_operation_async(
+        self, callee: Callable, retry_settings: Optional[RetrySettings] = None, *args, **kwargs
+    ):
         """WARNING: This API is experimental and could be changed.
         Special interface to execute a bunch of commands with session in a safe, retriable way.
 

ydb/retries.py

@@ -158,4 +158,4 @@ async def retry_operation_async(callee, retry_settings=None, *args, **kwargs):
             try:
                 return await next_opt.result
             except BaseException as e:  # pylint: disable=W0703
-                next_opt.set_exception(e)
+                next_opt.set_exception(e)