-
Notifications
You must be signed in to change notification settings - Fork 171
Fix TorchScript warning "doesn't support instance-level annotations" #437
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
Fix TorchScript warning "doesn't support instance-level annotations" #437
Conversation
|
See also pytorch/pytorch#89064 |
|
Tell me when you want to merge |
|
Thanks @mariogeiger --- ideally just want to see the tests pass, and if anyone else wants to take a quick look to make sure I didn't make any typos / logical errors in this, since it touches some of the compilation logic that underpins most of the library. |
|
Looks good from my side. Thanks for cleaning up the Torch2 stuff |
|
There is a problem with the tests |
|
Why does this file exist? https://github.com/e3nn/e3nn/blob/main/examples/conftest.py |
|
No idea, does it help it you delete it? |
|
Let's find out 😄 |
|
I suspect it's due to a combination of that file and the fact that the GitHub actions are set up run any tests anywhere in the repo: instead of only in the (Either way seem to be running fine now) |
|
I remember the examples were failing because of torch-cluster, torch-sparse… import issues when I was testing locally. Hopefully this also fixed that Update: Yup it did ! |
|
For the Torch 2 test that is failing its running alright locally when I do so not sure what's happening there |
|
Is there a way to just not run tests or flake8 on anything in |
|
Either way, it does look like all failures are unrelated to this PR, so I think we're good to merge, thanks! |
|
Do you mind doing a |
|
Ok let me merge and feel free to make another PR to fix this test problem |
|
Great, thanks guys! |
Description
In recent PyTorch versions, this warning has started coming up a lot:
How Has This Been Tested?
All unit tests pass locally, with GPU.
Checklist: