chore: blackify (#6146)

Co-authored-by: Jina Dev Bot <[email protected]>

jina/__init__.py

@@ -35,7 +35,7 @@ def _ignore_google_warnings():
         'ignore',
         category=DeprecationWarning,
         message='Deprecated call to `pkg_resources.declare_namespace(\'google\')`.',
-        append=True
+        append=True,
     )
 
 

jina/clients/__init__.py

@@ -1,4 +1,5 @@
 """Module wrapping the Client of Jina."""
+
 import argparse
 from typing import TYPE_CHECKING, List, Optional, Union, overload
 

jina/clients/base/__init__.py

@@ -1,4 +1,5 @@
 """Module containing the Base Client for Jina."""
+
 import abc
 import argparse
 import inspect
@@ -48,9 +49,11 @@ def __init__(
             os.unsetenv('https_proxy')
         self._inputs = None
         self._setup_instrumentation(
-            name=self.args.name
-            if hasattr(self.args, 'name')
-            else self.__class__.__name__,
+            name=(
+                self.args.name
+                if hasattr(self.args, 'name')
+                else self.__class__.__name__
+            ),
             tracing=self.args.tracing,
             traces_exporter_host=self.args.traces_exporter_host,
             traces_exporter_port=self.args.traces_exporter_port,
@@ -180,8 +183,7 @@ async def _get_results(
         on_error: Optional['CallbackFnType'] = None,
         on_always: Optional['CallbackFnType'] = None,
         **kwargs,
-    ):
-        ...
+    ): ...
 
     @abc.abstractmethod
     def _is_flow_ready(self, **kwargs) -> bool:

jina/clients/base/grpc.py

@@ -142,7 +142,9 @@ async def _get_results(
                                     compression=self.compression,
                                     **kwargs,
                                 )
-                                async for response in stream_rpc.stream_rpc_with_retry():
+                                async for (
+                                    response
+                                ) in stream_rpc.stream_rpc_with_retry():
                                     yield response
                             else:
                                 unary_rpc = UnaryRpc(
@@ -257,9 +259,7 @@ async def _get_streaming_results(
         req.header.exec_endpoint = on
         req.document_cls = inputs.__class__
         req.data.doc = inputs
-        async for response in self.stream_doc_endpoint(
-            request=req, timeout=timeout
-        ):
+        async for response in self.stream_doc_endpoint(request=req, timeout=timeout):
             yield return_type.from_protobuf(response.document)
 
 

jina/clients/base/retry.py

@@ -33,13 +33,14 @@ def _raise_last_attempt(err, attempt):
         raise err
 
 
-def sync_wait_or_raise_err(attempt: int,
-                           err: Exception,
-                           max_attempts: float,
-                           backoff_multiplier: float,
-                           initial_backoff: float,
-                           max_backoff: float,
-                           ):
+def sync_wait_or_raise_err(
+    attempt: int,
+    err: Exception,
+    max_attempts: float,
+    backoff_multiplier: float,
+    initial_backoff: float,
+    max_backoff: float,
+):
     """
     Accepts retry parameters and the underlying. The error is raised if the max_attempts has been reached otherwise the
     method waits based on the backoff calculations.
@@ -53,16 +54,18 @@ def sync_wait_or_raise_err(attempt: int,
     if attempt == max_attempts:
         _raise_last_attempt(err, attempt)
     else:
-        time.sleep(_wait_time(attempt, backoff_multiplier, initial_backoff, max_backoff))
+        time.sleep(
+            _wait_time(attempt, backoff_multiplier, initial_backoff, max_backoff)
+        )
 
 
 async def wait_or_raise_err(
-        attempt: int,
-        err: Exception,
-        max_attempts: float,
-        backoff_multiplier: float,
-        initial_backoff: float,
-        max_backoff: float,
+    attempt: int,
+    err: Exception,
+    max_attempts: float,
+    backoff_multiplier: float,
+    initial_backoff: float,
+    max_backoff: float,
 ):
     """
     Accepts retry parameters and the underlying. The error is raised if the max_attempts has been reached otherwise the
@@ -78,7 +81,9 @@ async def wait_or_raise_err(
     if attempt == max_attempts:
         _raise_last_attempt(err, attempt)
     else:
-        await asyncio.sleep(_wait_time(attempt, backoff_multiplier, initial_backoff, max_backoff))
+        await asyncio.sleep(
+            _wait_time(attempt, backoff_multiplier, initial_backoff, max_backoff)
+        )
 
 
 def _wait_time(attempt, backoff_multiplier, initial_backoff, max_backoff):

jina/clients/base/websocket.py

@@ -1,4 +1,5 @@
 """A module for the websockets-based Client for Jina."""
+
 import asyncio
 from contextlib import AsyncExitStack
 from typing import TYPE_CHECKING, Dict, Optional, Tuple
@@ -131,9 +132,9 @@ async def _get_results(
                 )
             )
 
-            request_buffer: Dict[
-                str, asyncio.Future
-            ] = dict()  # maps request_ids to futures (tasks)
+            request_buffer: Dict[str, asyncio.Future] = (
+                dict()
+            )  # maps request_ids to futures (tasks)
 
             def _result_handler(result):
                 return result

jina/clients/mixin.py

@@ -401,6 +401,7 @@ async def _get_results(*args, **kwargs):
             inferred_return_type = return_type
             if docarray_v2:
                 from docarray import DocList
+
                 if not issubclass(return_type, DocList):
                     is_singleton = True
                     inferred_return_type = DocList[return_type]
@@ -530,6 +531,7 @@ async def post(
             is_singleton = False
             if docarray_v2:
                 from docarray import DocList
+
                 if issubclass(return_type, DocList):
                     result.document_array_cls = return_type
                 else:

jina/clients/request/helper.py

@@ -1,4 +1,5 @@
 """Module for helper functions for clients."""
+
 from typing import Optional, Tuple
 
 from jina._docarray import Document, DocumentArray, docarray_v2

jina/excepts.py

@@ -1,4 +1,5 @@
 """This modules defines all kinds of exceptions raised in Jina."""
+
 from typing import List, Optional, Set, Union
 
 import grpc.aio

jina/exporter.py

@@ -22,7 +22,9 @@ def export_kubernetes(args):
             output_base_path=args.outpath, k8s_namespace=args.k8s_namespace
         )
     else:
-        raise NotImplementedError(f'Object of class {obj.__class__.__name__} cannot be exported to Kubernetes')
+        raise NotImplementedError(
+            f'Object of class {obj.__class__.__name__} cannot be exported to Kubernetes'
+        )
 
 
 def export_docker_compose(args):
@@ -40,7 +42,9 @@ def export_docker_compose(args):
             output_path=args.outpath, network_name=args.network_name
         )
     else:
-        raise NotImplementedError(f'Object of class {obj.__class__.__name__} cannot be exported to Docker Compose')
+        raise NotImplementedError(
+            f'Object of class {obj.__class__.__name__} cannot be exported to Docker Compose'
+        )
 
 
 def export_flowchart(args):
@@ -59,6 +63,7 @@ def export_schema(args):
     :param args: args from CLI
     """
     from jina import __version__
+
     if args.yaml_path:
         dump_api = api_to_dict()
         for yp in args.yaml_path:

jina/helper.py

@@ -1666,4 +1666,5 @@ def _telemetry():
 
 def is_generator(func):
     import inspect
+
     return inspect.isgeneratorfunction(func) or inspect.isasyncgenfunction(func)

jina/importer.py

@@ -39,18 +39,20 @@ def __init__(
 
     def __enter__(self):
         return self
-    
+
     def _check_v(self, v, missing_module):
         if (
-                v.strip()
-                and not v.startswith('#')
-                and v.startswith(missing_module)
-                and ':' in v
-            ):
+            v.strip()
+            and not v.startswith('#')
+            and v.startswith(missing_module)
+            and ':' in v
+        ):
             return True
 
     def _find_missing_module_in_extra_req(self, missing_module):
-        with open(os.path.join(__resources_path__, 'extra-requirements.txt'), encoding='utf-8') as fp:
+        with open(
+            os.path.join(__resources_path__, 'extra-requirements.txt'), encoding='utf-8'
+        ) as fp:
             for v in fp:
                 if self._check_v(v, missing_module):
                     missing_module, install_tags = v.split(':')
@@ -63,7 +65,6 @@ def _find_missing_module(self, exc_val):
         missing_module = self._find_missing_module_in_extra_req(missing_module)
         return missing_module
 
-
     def _err_msg(self, exc_val, missing_module):
         if self._tags:
             from jina.helper import colored
@@ -84,7 +85,7 @@ def _err_msg(self, exc_val, missing_module):
         else:
             err_msg = f'{exc_val.msg}'
         return err_msg
-    
+
     def _log_critical(self, err_msg):
         if self._verbose and self._logger:
             self._logger.critical(err_msg)
@@ -95,7 +96,7 @@ def _log_warning(self, err_msg):
         if self._verbose and self._logger:
             self._logger.warning(err_msg)
             if self._help_text:
-                self._logger.info(self._help_text)   
+                self._logger.info(self._help_text)
 
     def _raise_or_supress(self, err_msg, exc_val):
         if self._verbose and not self._logger:
@@ -107,14 +108,12 @@ def _raise_or_supress(self, err_msg, exc_val):
             self._log_warning(err_msg)
             return True  # suppress the error
 
-
     def __exit__(self, exc_type, exc_val, traceback):
         if exc_type != ModuleNotFoundError:
             return
         missing_module = self._find_missing_module(exc_val)
         err_msg = self._err_msg(exc_val, missing_module)
         return self._raise_or_supress(err_msg, exc_val)
-
 
 
 def _path_import(absolute_path: str):

jina/jaml/__init__.py

@@ -680,9 +680,7 @@ def load_config(
         :return: :class:`JAMLCompatible` object
         """
         if runtime_args:
-            kwargs[
-                'runtimes_args'
-            ] = (
+            kwargs['runtimes_args'] = (
                 dict()
             )  # when we have runtime args it is needed to have an empty runtime args session in the yam config
 
@@ -741,9 +739,11 @@ def _delitem(
                 _extra_search_paths = extra_search_paths or []
                 load_py_modules(
                     no_tag_yml,
-                    extra_search_paths=(_extra_search_paths + [os.path.dirname(s_path)])
-                    if s_path
-                    else _extra_search_paths,
+                    extra_search_paths=(
+                        (_extra_search_paths + [os.path.dirname(s_path)])
+                        if s_path
+                        else _extra_search_paths
+                    ),
                 )
 
             from jina.enums import DeploymentRoleType

jina/orchestrate/deployments/__init__.py

@@ -605,9 +605,11 @@ def _get_connection_list_for_flow(self) -> List[str]:
             # there is no head, add the worker connection information instead
             ports = self.ports
             hosts = [
-                __docker_host__
-                if host_is_local(host) and in_docker() and self._is_docker
-                else host
+                (
+                    __docker_host__
+                    if host_is_local(host) and in_docker() and self._is_docker
+                    else host
+                )
                 for host in self.hosts
             ]
             return [
@@ -1133,9 +1135,11 @@ def start(self) -> 'Deployment':
                 deployment_args=self.args,
                 args=self.pod_args['pods'][shard_id],
                 head_pod=self.head_pod,
-                name=f'{self.name}-replica-set-{shard_id}'
-                if num_shards > 1
-                else f'{self.name}-replica-set',
+                name=(
+                    f'{self.name}-replica-set-{shard_id}'
+                    if num_shards > 1
+                    else f'{self.name}-replica-set'
+                ),
             )
             self.enter_context(self.shards[shard_id])
 

jina/orchestrate/deployments/config/docker_compose.py

@@ -275,13 +275,13 @@ def __init__(
                     shard_id=i,
                     common_args=self.args,
                     service_args=args,
-                    pod_type=PodRoleType.WORKER
-                    if name != 'gateway'
-                    else PodRoleType.GATEWAY,
+                    pod_type=(
+                        PodRoleType.WORKER if name != 'gateway' else PodRoleType.GATEWAY
+                    ),
                     jina_deployment_name=self.name,
-                    deployments_addresses=self.deployments_addresses
-                    if name == 'gateway'
-                    else None,
+                    deployments_addresses=(
+                        self.deployments_addresses if name == 'gateway' else None
+                    ),
                 )
             )
 
@@ -342,9 +342,7 @@ def _get_services_args(self, args):
             uses_before_cargs.pod_role = PodRoleType.WORKER
             uses_before_cargs.polling = None
             parsed_args['uses_before_service'] = uses_before_cargs
-            parsed_args[
-                'head_service'
-            ].uses_before_address = (
+            parsed_args['head_service'].uses_before_address = (
                 f'{to_compatible_name(uses_before_cargs.name)}:{uses_before_cargs.port}'
             )
         if uses_after and shards > 1:
@@ -366,9 +364,7 @@ def _get_services_args(self, args):
             uses_after_cargs.pod_role = PodRoleType.WORKER
             uses_after_cargs.polling = None
             parsed_args['uses_after_service'] = uses_after_cargs
-            parsed_args[
-                'head_service'
-            ].uses_after_address = (
+            parsed_args['head_service'].uses_after_address = (
                 f'{to_compatible_name(uses_after_cargs.name)}:{uses_after_cargs.port}'
             )