search

patrick-kidger / torchtyping

Type annotations and dynamic checking for a tensor's shape, dtype, names, etc.
Apache License 2.0
1.41k stars 34 forks source link

mypy integration #1

Closed patrick-kidger closed 3 years ago

patrick-kidger commented 3 years ago

torchtyping currently integrates with typeguard. Integration with mypy would be good to have as well.

It certainly won't be as strong -- there's no way that mypy is powerful enough to can catch shape/dtype errors -- but it's worth thinking about.