NVIDIA
[Feature Request] Python type annotation
Are there any plans for adding type hints to the Python bindings? (If not, I'm willing to submit a PR :smile: )
Currently our library is not set up to accept PR from github but we appreciate your feedback. I will create an internal ticket to keep track of this feature request.
Hi @EarlMilktea, could you be more explicitly on which APIs you need for type hints? Note that the low-level bindings are auto-generated Cython code (we do not modify them manually) and already use Cython static typing (instead of the Pure Python Mode). Cython's .pxd/.pyx files aren't really fully compatible with PEP-484 type hints. For example, cython/cython#4756 is one relevant issue.
For high-level pythonic APIs I agree type hints can be valuable, though some of the functions are hard to annotate (ex: contract accepts multiple array types, making typing impossible).
I don't have specific requests for details.
Let me just say that I submitted this issue with numpy.typing in mind: I know it cannot be exhaustive (ex. undocumented/internal modules can be untyped) or complete (ex. return type can be Any if too complex/array dtype can be missing) in principle.
Thank you in advance!