fix(request_generator): send index request in index mode

gnes-ai · Aug 13, 2019 · fd76aa7 · fd76aa7
1 parent 64163cb
commit fd76aa7
Showing 1 changed file with 1 addition and 1 deletion.
diff --git 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):