-
Notifications
You must be signed in to change notification settings - Fork 3.5k
New issue
Have a question about this project? Sign up for a free GitHub account to open an issue and contact its maintainers and the community.
By clicking “Sign up for GitHub”, you agree to our terms of service and privacy statement. We’ll occasionally send you account related emails.
Already on GitHub? Sign in to your account
[TVMScript][Fix] Add type hints for more uncovered cases #9505
Conversation
Thank you so much for acting swiftly on this!! BTW, would you mind also adding T.exp in the unittest? |
@shingjan please address comments above |
b38fccf
to
96740bc
Compare
@vinx13 Comments addressed. I fixed the CI issue by disabling the mypy check on slice index of that two lines. |
The following two errors should be fixed rather than skipped (if possible)
|
It is an issue of mypy that can't be fixed - it requires slice must be constructed with int rather than custom type |
There are still some CI error. Please make sure it can pass locally |
Shall we merge this PR? |
* add support for prevously uncovered cases * remove PrimExpr import * add exp test and mypy ignore * disable ling too long * resolve long line * nit * add dtype to unary ops
* add support for prevously uncovered cases * remove PrimExpr import * add exp test and mypy ignore * disable ling too long * resolve long line * nit * add dtype to unary ops
* add support for prevously uncovered cases * remove PrimExpr import * add exp test and mypy ignore * disable ling too long * resolve long line * nit * add dtype to unary ops
* add support for prevously uncovered cases * remove PrimExpr import * add exp test and mypy ignore * disable ling too long * resolve long line * nit * add dtype to unary ops
* add support for prevously uncovered cases * remove PrimExpr import * add exp test and mypy ignore * disable ling too long * resolve long line * nit * add dtype to unary ops
This PR intends to add type hints for some uncovered cases in TVMScript.
cc: @junrushao1994 @vinx13