From 095e7020013d7a7e251f95f607813b200cfb1634 Mon Sep 17 00:00:00 2001 From: John Colby Date: Fri, 24 Apr 2020 12:51:04 -0700 Subject: [PATCH] Fix input argument encoding Fixes #909 --- mms/model_service_worker.py | 9 ++++++--- 1 file changed, 6 insertions(+), 3 deletions(-) diff --git a/mms/model_service_worker.py b/mms/model_service_worker.py index 4b8101c09..f89bbc20f 100644 --- a/mms/model_service_worker.py +++ b/mms/model_service_worker.py @@ -232,9 +232,12 @@ def run_server(self): sock_type = args.sock_type host = args.host port = args.port - model_req["handler"] = args.handler.encode('utf-8') - model_req["modelPath"] = args.model_path.encode('utf-8') - model_req["modelName"] = args.model_name.encode('utf-8') + if args.handler is not None: + model_req["handler"] = args.handler.encode('utf-8') + if args.model_path is not None: + model_req["modelPath"] = args.model_path.encode('utf-8') + if args.model_name is not None: + model_req["modelName"] = args.model_name.encode('utf-8') worker = MXNetModelServiceWorker(sock_type, socket_name, host, port, model_req, args.preload_model, args.tmp_dir)