Any plan for improving type hint

[noTmiHoYo]

I am using vscode with pylance. Autocomplete and typehint almost won't work..

Are there any one are current working on this? Or should I create a PR to fix some of these problems?

Several examples.

Example 1

PyLance cannot figure out what does VectorType stand for..

Example 2

possible fix for example 1&2: I found these problems are fixed after I add some import statement in __init__.py.

Example 3

ti.math.vec2(1, 2).x is valid. But no completion is provided.
possible fix: provide special type hint for vec2, vec3, vec4, ivec2, ivec3, etc..

Example 4

though ti.math.vec2 can be used as type when define a taichi struct, but it cannot be used in isinstance(ti.math.vec2(1, 2), ti.math.vec2) because it is not a real python type.
possible fix: How about make ti.types types real types inside C code ?

Example #5

when i try to access a member variable of a object of classes decorated by @ti.dataclass, no auto completion is provided.
possible fix: PEP 681

[neozhaoliang]

Hi, you are welcome to contribute! The suggestions for these examples are very valuable.

Let me explain why vec2.x does not have an auto completion: this attribute is added to the Matrix class via a decorator: https://github.com/taichi-dev/taichi/blob/master/python/taichi/lang/matrix.py#L370
and hence the code autocompletion plugin needs to "run" this decorator to see it, which is infeasible.