chore(style): reformatted by jina-dev-bot

jina/drivers/__init__.py

@@ -121,10 +121,10 @@ def _get_parameter(self, key: str, default: Any):
         if getattr(self, 'queryset', None):
             for q in self.queryset:
                 if (
-                        not q.disabled
-                        and self.__class__.__name__ == q.name
-                        and q.priority > self._priority
-                        and key in q.parameters
+                    not q.disabled
+                    and self.__class__.__name__ == q.name
+                    and q.priority > self._priority
+                    and key in q.parameters
                 ):
                     ret = q.parameters[key]
                     return dict(ret) if isinstance(ret, Struct) else ret
@@ -282,8 +282,7 @@ def __getstate__(self) -> Dict[str, Any]:
 
 
 class RecursiveMixin(BaseDriver):
-    """A mixin to traverse a set of Documents with a specific path. to be mixed in with :class:`BaseRecursiveDriver`
-    """
+    """A mixin to traverse a set of Documents with a specific path. to be mixed in with :class:`BaseRecursiveDriver`"""
 
     @property
     def docs(self):
@@ -297,23 +296,23 @@ def docs(self):
             return self.req.docs
 
     def _apply_root(
-            self,
-            docs: 'DocumentSet',
-            field: str,
-            *args,
-            **kwargs,
+        self,
+        docs: 'DocumentSet',
+        field: str,
+        *args,
+        **kwargs,
     ) -> None:
         return self._apply_all(docs, None, field, *args, **kwargs)
 
     # TODO(Han): probably want to publicize this, as it is not obvious for driver
     #  developer which one should be inherited
     def _apply_all(
-            self,
-            docs: 'DocumentSet',
-            context_doc: 'Document',
-            field: str,
-            *args,
-            **kwargs,
+        self,
+        docs: 'DocumentSet',
+        context_doc: 'Document',
+        field: str,
+        *args,
+        **kwargs,
     ) -> None:
         """Apply function works on a list of docs, modify the docs in-place
 
@@ -364,11 +363,11 @@ def _traverse_rec(self, docs, parent_doc, parent_edge_type, path, *args, **kwarg
 
 class FastRecursiveMixin:
     """
-     The optimized version of :class:`RecursiveMixin`, to be mixed in with :class:`BaseRecursiveDriver`
-     it uses :meth:`traverse` in :class:`DocumentSet` and yield much better performance for index and encode drivers.
+    The optimized version of :class:`RecursiveMixin`, to be mixed in with :class:`BaseRecursiveDriver`
+    it uses :meth:`traverse` in :class:`DocumentSet` and yield much better performance for index and encode drivers.
 
-     .. seealso::
-        https://github.com/jina-ai/jina/issues/1932
+    .. seealso::
+       https://github.com/jina-ai/jina/issues/1932
 
     """
 
@@ -400,9 +399,7 @@ class BaseRecursiveDriver(BaseDriver):
     It is intended to be mixed in with either :class:`FastRecursiveMixin` or :class:`RecursiveMixin`
     """
 
-    def __init__(self,
-                 traversal_paths: Tuple[str] = ('c', 'r'),
-                 *args, **kwargs):
+    def __init__(self, traversal_paths: Tuple[str] = ('c', 'r'), *args, **kwargs):
         """Initialize a :class:`BaseRecursiveDriver`
 
         :param traversal_paths: Describes the leaves of the document tree on which _apply_all are called
@@ -422,9 +419,13 @@ class BaseExecutableDriver(BaseRecursiveDriver):
     This is done by :func:`attach`. Note that a deserialized :class:`BaseDriver` from file is always unattached.
     """
 
-    def __init__(self, executor: Optional[str] = None,
-                 method: Optional[str] = None,
-                 *args, **kwargs):
+    def __init__(
+        self,
+        executor: Optional[str] = None,
+        method: Optional[str] = None,
+        *args,
+        **kwargs,
+    ):
         """Initialize a :class:`BaseExecutableDriver`
 
         :param executor: the name of the sub-executor, only necessary when :class:`jina.executors.compound.CompoundExecutor` is used
@@ -452,8 +453,8 @@ def exec_fn(self) -> Callable:
         :return: the Callable to execute in the driver
         """
         if (
-                not self.msg.is_error
-                or self.runtime.args.on_error_strategy < OnErrorStrategy.SKIP_EXECUTOR
+            not self.msg.is_error
+            or self.runtime.args.on_error_strategy < OnErrorStrategy.SKIP_EXECUTOR
         ):
             return self._exec_fn
         else:
@@ -473,7 +474,7 @@ def attach(self, executor: 'AnyExecutor', *args, **kwargs) -> None:
             else:
                 for c in executor.components:
                     if any(
-                            t.__name__ == self._executor_name for t in type.mro(c.__class__)
+                        t.__name__ == self._executor_name for t in type.mro(c.__class__)
                     ):
                         self._exec = c
                         break