fix ut

3a1cb099 · MRXLT · c8474c43 · 3a1cb099
显示空白变更内容
内联并排

Showing with 1 addition and 2 deletion

python/examples/grpc_impl_example/fit_a_line/test_timeout_client.py ...mples/grpc_impl_example/fit_a_line/test_timeout_client.py +1 -2

未找到文件。
--- 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"])