From edcf113205375e71d625dddce8b490fa1ca10222 Mon Sep 17 00:00:00 2001 From: gongweibao Date: Sun, 14 Jun 2020 12:25:05 +0000 Subject: [PATCH] add get_config test=develop --- python/paddle_serving_client/__init__.py | 5 +++-- python/paddle_serving_server/__init__.py | 2 +- 2 files changed, 4 insertions(+), 3 deletions(-) diff --git a/python/paddle_serving_client/__init__.py b/python/paddle_serving_client/__init__.py index 7a36f370..09dd1ae1 100644 --- a/python/paddle_serving_client/__init__.py +++ b/python/paddle_serving_client/__init__.py @@ -26,7 +26,8 @@ import grpc from .proto import multi_lang_general_model_service_pb2 sys.path.append( os.path.join(os.path.abspath(os.path.dirname(__file__)), 'proto')) -from .proto import grpc_pb2 as grpc_pb2 +from .proto import multi_lang_general_model_service_pb2 as pb2 +from .proto import multi_lang_general_model_service_pb2_grpc as grpc_pb2 int_type = 0 float_type = 1 @@ -395,7 +396,7 @@ class MultiLangClient(object): self._parse_model_config(proto_txt) def _load_client_config(self, stub): - req= grpc_pb2.ServingConfig() + req= pb2.ServingConfig() self._config = self.stub_.get_client_proto_text(req) self._parse_model_config(config.proto_txt) diff --git a/python/paddle_serving_server/__init__.py b/python/paddle_serving_server/__init__.py index fbcc138d..7c8b525a 100644 --- a/python/paddle_serving_server/__init__.py +++ b/python/paddle_serving_server/__init__.py @@ -441,7 +441,7 @@ class Server(object): class MultiLangServerService( - pb2_grpc.MultiLangGeneralModelService): + grpc_pb2.MultiLangGeneralModelService): def __init__(self, model_config_path, endpoints): from paddle_serving_client import Client -- GitLab