fix(client): fix check input function

jina-ai · May 28, 2020 · 76bd4e6 · 76bd4e6
1 parent e1e42f0
commit 76bd4e6
Showing 1 changed file with 3 additions and 0 deletions.
diff --git a/jina/clients/python/__init__.py b/jina/clients/python/__init__.py
@@ -70,6 +70,9 @@ def check_input(input_fn: Union[Iterator['jina_pb2.Document'], Iterator[bytes],
         :param input_fn: the input function
         :param input_type: if the input data is in protobuf Document format, or in raw bytes, or data uri
         """
+        if hasattr(input_fn, '__call__'):
+            input_fn = input_fn()
+
         kwargs = {'data': input_fn}
 
         try: