Merge branch 'fix-dbatch' of https://github.com/jina-ai/jina into fix…

…-dbatch

Signed-off-by: Joan Fontanals Martinez <[email protected]>

jina/serve/runtimes/worker/batch_queue.py

@@ -132,6 +132,11 @@ async def _await_then_flush(self) -> None:
                         raise ValueError(
                             f'Dynamic Batching requires input size to equal output size. Expected output size {input_len_before_call}, but got {len(self._big_doc)}'
                         )
+                elif return_docs is None:
+                    if not len(self._big_doc) == input_len_before_call:
+                        raise ValueError(
+                            f'Dynamic Batching requires input size to equal output size. Expected output size {input_len_before_call}, but got {len(self._big_doc)}'
+                        )
                 else:
                     array_name = 'DocumentArray' if not docarray_v2 else 'DocList'
                     raise TypeError(