chore: add proto_path $GOPATH/pkg/mod

fal-ai · Dec 22, 2023 · aec894f · aec894f
1 parent 1af1d26
commit aec894f
Showing 1 changed file with 3 additions and 1 deletion.
diff --git a/tools/regen_grpc.py b/tools/regen_grpc.py
@@ -1,6 +1,7 @@
 from __future__ import annotations
 
 import ast
+import os
 import shutil
 import subprocess
 import sys
@@ -73,7 +74,8 @@ def regen_grpc(
             sys.executable,
             "-m",
             "grpc_tools.protoc",
-            "--proto_path=.",
+            "-I=.",
+            f"-I={os.getenv('GOPATH')}/pkg/mod",
             "--python_out=.",
             "--grpc_python_out=.",
             "--mypy_out=.",