feat: add mypy interface files to aid with type checking

substrait-io · Mar 27, 2024 · 712040c · 712040c
1 parent 51ec42a
commit 712040c
Show file tree

Hide file tree

Showing 15 changed files with 6,571 additions and 3 deletions.
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