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

TensorType now inherits from torch.Tensor so that IDE lookup+error messages work #20

Closed patrick-kidger closed 3 years ago

patrick-kidger commented 3 years ago

This should close #18.

Also updated some pre-commit hooks that were giving me trouble for some reason.