feat: wait until grpc channel is ready before warmup

jina-ai · Jan 12, 2023 · 34d7d4b · 34d7d4b
1 parent 367a115
commit 34d7d4b
Showing 1 changed file with 1 addition and 0 deletions.
diff --git a/jina/serve/networking.py b/jina/serve/networking.py
@@ -1125,6 +1125,7 @@ async def warmup(
 
         async def task_wrapper(target_warmup_responses, target, channel):
             try:
+                await channel.channel_ready()
                 stub = jina_pb2_grpc.JinaInfoRPCStub(channel=channel)
                 call_result = stub._status(
                     request=jina_pb2.google_dot_protobuf_dot_empty__pb2.Empty(),