From fd76aa79b75ccb68fa3c345306d148cc21de44a7 Mon Sep 17 00:00:00 2001 From: Jem Date: Tue, 13 Aug 2019 11:03:07 +0800 Subject: [PATCH] fix(request_generator): send index request in index mode --- gnes/client/cli.py | 2 +- 1 file changed, 1 insertion(+), 1 deletion(-) diff --git a/gnes/client/cli.py b/gnes/client/cli.py index 6a8abade..2ed32016 100644 --- a/gnes/client/cli.py +++ b/gnes/client/cli.py @@ -45,7 +45,7 @@ def __init__(self, args): resp = list(stub.StreamCall(RequestGenerator.train(all_bytes, args.batch_size)))[-1] print(resp) elif args.mode == 'index': - resp = list(stub.StreamCall(RequestGenerator.train(all_bytes, args.batch_size)))[-1] + resp = list(stub.StreamCall(RequestGenerator.index(all_bytes, args.batch_size)))[-1] print(resp) elif args.mode == 'query': for idx, q in enumerate(all_bytes):