feat: add mypy and mypy grpc suffixes

cpcloud · Nov 26, 2021 · e5a4b9e · e5a4b9e
1 parent 37f4481
commit e5a4b9e
Showing 1 changed file with 2 additions and 2 deletions.
diff --git a/protoletariat/__main__.py b/protoletariat/__main__.py
@@ -50,8 +50,8 @@ def _echo(_: Path, code: str) -> None:
     "--module-suffixes",
     type=str,
     multiple=True,
-    default=["_pb2.py", "_pb2_grpc.py"],
-    help="Suffixes of Python modules to process",
+    default=["_pb2.py", "_pb2.pyi", "_pb2_grpc.py", "_pb2_grpc.pyi"],
+    help="Suffixes of Python/mypy modules to process",
     show_default=True,
 )
 @click.pass_context