-
Notifications
You must be signed in to change notification settings - Fork 3.7k
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
[Type Hints] nn.GNNExplainer
#5716
[Type Hints] nn.GNNExplainer
#5716
Conversation
Codecov Report
@@ Coverage Diff @@
## master #5716 +/- ##
=========================================
Coverage ? 83.97%
=========================================
Files ? 349
Lines ? 19407
Branches ? 0
=========================================
Hits ? 16297
Misses ? 3110
Partials ? 0
📣 We’re building smart automated test selection to slash your CI/CD build times. Learn more |
There was a problem hiding this comment.
Choose a reason for hiding this comment
The reason will be displayed to describe this comment to others. Learn more.
torch.jit.export
works here. Can we also add type hints to the function inside GNNExplainer
?
Hi @rusty1s, I already did. I added type hints to all methods in the class Am I missing something? |
Ah, sorry. Overlooked that single file, my bad. |
No problem! I'll resolve the Changelog conflicts and this should be ready to merge. |
@rusty1s Please note that I am using `torch.jit.export()` instead of `torch.jit.script()` for the tests. Can you please confirm if this is ok? Co-authored-by: Matthias Fey <matthias.fey@tu-dortmund.de>
@rusty1s Please note that I am using
torch.jit.export()
instead oftorch.jit.script()
for the tests. Can you please confirm if this is ok?