Skip to content
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

Move GraphMultisetTransformer to nn.aggr package #4973

Merged
merged 4 commits into from
Jul 14, 2022

Conversation

Padarn
Copy link
Contributor

@Padarn Padarn commented Jul 13, 2022

Addresses #4712

@Padarn Padarn requested review from rusty1s and lightaime July 13, 2022 00:48
@Padarn
Copy link
Contributor Author

Padarn commented Jul 13, 2022

Hey guys this one pulled up a small issue (maybe there is a simple fix I don't see).

The GraphMultisetTransformer imports GCNConv which is part of the nn.conv package.. but the nn.aggr package is also imported by the nn.conv package in MessagPassing.

Fixes I could think of

  1. move GraphMultisetTransformer into nn.conv
  2. move MessagPassing into its own package

Any other ideas? Given how GraphMultisetTransformer works (it has optional edge_index) I'm not very sure where it best lives.

WDYT @lightaime @rusty1s

@rusty1s
Copy link
Member

rusty1s commented Jul 13, 2022

@Padarn You should be able to import nn.conv.GCNConv inside GraphMultisetTransformer.__init__

@Padarn
Copy link
Contributor Author

Padarn commented Jul 13, 2022 via email

@codecov
Copy link

codecov bot commented Jul 14, 2022

Codecov Report

Merging #4973 (6c15ce1) into master (a7224df) will increase coverage by 0.00%.
The diff coverage is 100.00%.

@@           Coverage Diff           @@
##           master    #4973   +/-   ##
=======================================
  Coverage   82.78%   82.78%           
=======================================
  Files         330      330           
  Lines       17970    17973    +3     
=======================================
+ Hits        14876    14879    +3     
  Misses       3094     3094           
Impacted Files Coverage Δ
torch_geometric/nn/aggr/__init__.py 100.00% <100.00%> (ø)
torch_geometric/nn/aggr/base.py 93.61% <100.00%> (ø)
torch_geometric/nn/aggr/gmt.py 100.00% <100.00%> (ø)
torch_geometric/nn/glob/__init__.py 100.00% <100.00%> (ø)

Continue to review full report at Codecov.

Legend - Click here to learn more
Δ = absolute <relative> (impact), ø = not affected, ? = missing data
Powered by Codecov. Last update a7224df...6c15ce1. Read the comment docs.

Copy link
Contributor

@lightaime lightaime left a comment

Choose a reason for hiding this comment

The reason will be displayed to describe this comment to others. Learn more.

Thanks! LGTM!

@Padarn Padarn merged commit 8804f65 into pyg-team:master Jul 14, 2022
@Padarn Padarn mentioned this pull request Jul 15, 2022
26 tasks
Sign up for free to join this conversation on GitHub. Already have an account? Sign in to comment
Projects
None yet
Development

Successfully merging this pull request may close these issues.

3 participants