From 7c882ba689757ee3069c1128d2b625970fad8d73 Mon Sep 17 00:00:00 2001 From: MRXLT Date: Tue, 27 Oct 2020 19:32:35 +0800 Subject: [PATCH] fix ut --- .../grpc_impl_example/fit_a_line/test_timeout_client.py | 6 +++--- python/paddle_serving_client/__init__.py | 2 +- 2 files changed, 4 insertions(+), 4 deletions(-) diff --git a/python/examples/grpc_impl_example/fit_a_line/test_timeout_client.py b/python/examples/grpc_impl_example/fit_a_line/test_timeout_client.py index fcafaa34..5824c301 100644 --- a/python/examples/grpc_impl_example/fit_a_line/test_timeout_client.py +++ b/python/examples/grpc_impl_example/fit_a_line/test_timeout_client.py @@ -26,10 +26,10 @@ x = [ ] for i in range(3): fetch_map = client.predict(feed={"x": x}, fetch=["price"]) - if fetch_map["serving_grpc_error"] == 0: + if fetch_map["serving_status_code"] == 0: print(fetch_map) - elif fetch_map["serving_grpc_error"].code( + elif fetch_map["serving_status_code"].code( ) == grpc.StatusCode.DEADLINE_EXCEEDED: print('timeout') else: - print(fetch_map["serving_grpc_error"]) + print(fetch_map["serving_status_code"]) diff --git a/python/paddle_serving_client/__init__.py b/python/paddle_serving_client/__init__.py index 2ac32066..d350b584 100644 --- a/python/paddle_serving_client/__init__.py +++ b/python/paddle_serving_client/__init__.py @@ -643,7 +643,7 @@ class MultiLangClient(object): self.profile_.print_profile() return ret except grpc.RpcError as e: - return {"serving_grpc_error": e} + return {"serving_status_code": e.code()} else: req = self._pack_inference_request( feed, fetch, is_python=is_python, log_id=log_id) -- GitLab