ecmwf / anemoi-models

Apache License 2.0
34 stars 12 forks source link

Refactor node attributes #64

Closed JPXKQX closed 1 week ago

JPXKQX commented 1 month ago

This PR introduces the NamedNodesAttributes(nn.Module) class, designed to enhance the management of node attributes within graph neural networks. Key features include:

The main improvement is the ability to handle an arbitrary number of nodes. This development will facilitate extending AnemoiModelEncProcDec to other use cases, such as hierarchical models or multi-encoder/decoder.

codecov-commenter commented 1 month ago

Codecov Report

All modified and coverable lines are covered by tests :white_check_mark:

Project coverage is 99.85%. Comparing base (deba13a) to head (9e40145). Report is 2 commits behind head on develop.

Additional details and impacted files ```diff @@ Coverage Diff @@ ## develop #64 +/- ## ======================================== Coverage 99.84% 99.85% ======================================== Files 23 23 Lines 1301 1350 +49 ======================================== + Hits 1299 1348 +49 Misses 2 2 ```

:umbrella: View full report in Codecov by Sentry.
:loudspeaker: Have feedback on the report? Share it here.