Output .pyi files for protobufs.

Original change by @ckgresla in PR #246. PiperOrigin-RevId: 690816203
google-deepmind · Oct 29, 2024 · 0d883d3 · 0d883d3
1 parent 746e0e9
commit 0d883d3
Showing 1 changed file with 1 addition and 0 deletions.
diff --git a/setup.py b/setup.py
@@ -78,6 +78,7 @@ def run(self):
           '--proto_path={}'.format(grpc_protos_include),
           '--proto_path={}'.format(_ROOT_DIR),
           '--python_out={}'.format(_ROOT_DIR),
+          '--pyi_out={}'.format(_ROOT_DIR),
           '--grpc_python_out={}'.format(_ROOT_DIR),
           os.path.join(_ROOT_DIR, proto_path),
       ]