substrait-io · EpsilonPrime · Apr 1, 2024 · Mar 27, 2024
diff --git a/.gitignore b/.gitignore
@@ -142,6 +142,3 @@ src/substrait/_version.py
 .directory
 .gdb_history
 .DS_Store
-
-# Python interface files
-*.pyi
diff --git a/buf.gen.yaml b/buf.gen.yaml
@@ -1,4 +1,6 @@
 plugins:
   - name: python
     out: src/substrait/gen
+  - plugin: buf.build/community/nipunn1313-mypy:v3.5.0
+    out: src/substrait/gen
 version: v1
diff --git a/environment.yml b/environment.yml
@@ -7,6 +7,7 @@ dependencies:
   - pre-commit
   - protobuf = 3.20.1  # protobuf==3.20 C extensions aren't compatible with 3.19.4
   - protoletariat >= 2.0.0
+  - mypy-protobuf
   - pytest >= 7.0.0
   - python >= 3.8.1
   - setuptools >= 61.0.0

diff --git a/src/substrait/gen/__init__.pyi b/src/substrait/gen/__init__.pyi
diff --git a/src/substrait/gen/proto/__init__.pyi b/src/substrait/gen/proto/__init__.pyi
diff --git a/src/substrait/gen/proto/algebra_pb2.pyi b/src/substrait/gen/proto/algebra_pb2.pyi
diff --git a/src/substrait/gen/proto/capabilities_pb2.pyi b/src/substrait/gen/proto/capabilities_pb2.pyi
diff --git a/src/substrait/gen/proto/extended_expression_pb2.pyi b/src/substrait/gen/proto/extended_expression_pb2.pyi
diff --git a/src/substrait/gen/proto/extensions/__init__.pyi b/src/substrait/gen/proto/extensions/__init__.pyi
diff --git a/src/substrait/gen/proto/extensions/extensions_pb2.pyi b/src/substrait/gen/proto/extensions/extensions_pb2.pyi