fix: type hint

anna-charlotte · anna-charlotte · commit 4a3a290b0d2e · 2023-03-03T14:30:27.000+01:00
Signed-off-by: anna-charlotte &lt;charlotte.gerhaher@jina.ai&gt;
diff --git a/docarray/utils/apply.py b/docarray/utils/apply.py
@@ -141,7 +141,7 @@ def _map(
 
 def apply_batch(
     da: T,
-    func: Callable[[T], Union[T, T_doc]],
+    func: Union[Callable[[T], T], Callable[[T], T_doc]],
     batch_size: int,
     backend: str = 'thread',
     num_worker: Optional[int] = None,
