From 3483042906b8e2984d55ff112e12893b224801e2 Mon Sep 17 00:00:00 2001
From: kthui <18255193+kthui@users.noreply.github.com>
Date: Thu, 11 Jan 2024 12:01:27 -0800
Subject: [PATCH] Fix gRPC AsyncIO import

---
 src/python/library/tritonclient/grpc/aio/__init__.py | 11 +++++++++--
 1 file changed, 9 insertions(+), 2 deletions(-)

diff --git a/src/python/library/tritonclient/grpc/aio/__init__.py b/src/python/library/tritonclient/grpc/aio/__init__.py
index 3b4c949df..dd5c73fe3 100755
--- a/src/python/library/tritonclient/grpc/aio/__init__.py
+++ b/src/python/library/tritonclient/grpc/aio/__init__.py
@@ -29,15 +29,22 @@
 import base64
 import sys
 
+import grpc
 import rapidjson as json
 from google.protobuf.json_format import MessageToJson
 from tritonclient.grpc import *
+from tritonclient.grpc import model_config_pb2, service_pb2, service_pb2_grpc
 
 from ... import _auth as auth
-from ..._client import InferenceServerClientBase
 from ..._plugin import InferenceServerClientPlugin
 from ..._request import Request
-from .._utils import _get_inference_request, _grpc_compression_type
+from .._client import MAX_GRPC_MESSAGE_SIZE, InferenceServerClientBase, KeepAliveOptions
+from .._utils import (
+    _get_inference_request,
+    _grpc_compression_type,
+    raise_error,
+    raise_error_grpc,
+)
 
 
 class InferenceServerClient(InferenceServerClientBase):