From 1875eaffbdb01e88c319c5d965520ab2af071adf Mon Sep 17 00:00:00 2001 From: Han Xiao Date: Sat, 30 May 2020 00:01:33 +0200 Subject: [PATCH] feat(client): add input function sugar --- tests/test_cli.py | 8 +++++--- 1 file changed, 5 insertions(+), 3 deletions(-) diff --git a/tests/test_cli.py b/tests/test_cli.py index d4cb1830ceab1..0887ff36a74fa 100644 --- a/tests/test_cli.py +++ b/tests/test_cli.py @@ -3,11 +3,13 @@ import unittest from pathlib import Path +from pkg_resources import resource_filename + from jina.clients import py_client +from jina.clients.python.io import input_numpy from jina.flow import Flow -from jina.helloworld import download_data, input_fn +from jina.helloworld import download_data from jina.main.parser import set_hw_parser -from pkg_resources import resource_filename from tests import JinaTestCase @@ -59,7 +61,7 @@ def test_helloworld_flow(self): with f: py_client(host=f.host, port_grpc=f.port_grpc, - ).index(input_fn(targets['index']['filename']), batch_size=args.index_batch_size) + ).index(input_numpy(targets['index']['data']), batch_size=args.index_batch_size) if __name__ == '__main__':