fix: use unbound method to link to dry_run endpoint (#6143)

Signed-off-by: Joan Martinez <[email protected]>
Co-authored-by: Jina Dev Bot <[email protected]>

jina/serve/executors/__init__.py

@@ -400,7 +400,9 @@ def __init__(
         if __dry_run_endpoint__ not in self.requests:
             self.requests[
                 __dry_run_endpoint__
-            ] = _FunctionWithSchema.get_function_with_schema(self._dry_run_func)
+            ] = _FunctionWithSchema.get_function_with_schema(
+                self.__class__._dry_run_func
+            )
         else:
             self.logger.warning(
                 f' Endpoint {__dry_run_endpoint__} is defined by the Executor. Be aware that this endpoint is usually reserved to enable health checks from the Client through the gateway.'
@@ -409,7 +411,9 @@ def __init__(
         if type(self) == BaseExecutor:
             self.requests[
                 __default_endpoint__
-            ] = _FunctionWithSchema.get_function_with_schema(self._dry_run_func)
+            ] = _FunctionWithSchema.get_function_with_schema(
+                self.__class__._dry_run_func
+            )
 
         self._lock = contextlib.AsyncExitStack()
         try:

jina/serve/executors/decorators.py

@@ -5,7 +5,7 @@
 from pathlib import Path
 from typing import Callable, Dict, List, Optional, Sequence, Type, Union
 
-from jina._docarray import Document, DocumentArray, docarray_v2
+from jina._docarray import Document, DocumentArray
 from jina.constants import __cache_path__
 from jina.helper import is_generator, iscoroutinefunction
 from jina.importer import ImportExtensions