Type Hint

[Naveen934]

Hi, Type hinting is missing in some code, if add it will be easy to understand code structure. you can try with pyrefly

[lusxvr]

Very true! Feel free to open a PR to add it if you want to contribute :)