From 3a1cb09966b50b5df212de553f7cba2861a2a4c1 Mon Sep 17 00:00:00 2001 From: MRXLT Date: Wed, 28 Oct 2020 14:26:46 +0800 Subject: [PATCH] fix ut --- .../grpc_impl_example/fit_a_line/test_timeout_client.py | 3 +-- 1 file changed, 1 insertion(+), 2 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 5824c301..f90fab38 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 @@ -28,8 +28,7 @@ for i in range(3): fetch_map = client.predict(feed={"x": x}, fetch=["price"]) if fetch_map["serving_status_code"] == 0: print(fetch_map) - elif fetch_map["serving_status_code"].code( - ) == grpc.StatusCode.DEADLINE_EXCEEDED: + elif fetch_map["serving_status_code"] == grpc.StatusCode.DEADLINE_EXCEEDED: print('timeout') else: print(fetch_map["serving_status_code"]) -- GitLab