From a289607d5949f9398a737326492ae7776a1a50a2 Mon Sep 17 00:00:00 2001
From: Anita Ruangrotsakun <138700973+anitarua@users.noreply.github.com>
Date: Thu, 14 Nov 2024 10:51:22 -0800
Subject: [PATCH] fix: python protoc postprocessing script needs to account for
 import statements in pb2_grpc.py files too (#301)

---
 python/run-protoc.sh | 1 +
 1 file changed, 1 insertion(+)

diff --git a/python/run-protoc.sh b/python/run-protoc.sh
index 53134f8..54288ce 100755
--- a/python/run-protoc.sh
+++ b/python/run-protoc.sh
@@ -55,6 +55,7 @@ do
         # will work when installed, eg:
         # `import cacheclient_pb2` -> `from . import cacheclient_pb2`
         sed -i.old "s/^\(import $pb2_module_name as \)/from . \1/g" $src_path/*_pb2.py
+        sed -i.old "s/^\(import $pb2_module_name as \)/from . \1/g" $src_path/*_pb2_grpc.py
     done
 
     rm $src_path/*.old