facebookresearch / BenchMARL

A collection of MARL benchmarks based on TorchRL
https://benchmarl.readthedocs.io/
MIT License
216 stars 27 forks source link

[Model] GNN #30

Closed matteobettini closed 8 months ago

matteobettini commented 10 months ago

This PR aims at introducing a GNN model customizable via torch_geometric.

This is currently on hold because of:

https://github.com/pytorch/rl/issues/1576

codecov-commenter commented 9 months ago

Codecov Report

Attention: 63 lines in your changes are missing coverage. Please review.

Comparison is base (c6114e0) 89.47% compared to head (935e49e) 45.26%. Report is 2 commits behind head on main.

Files Patch % Lines
benchmarl/models/gnn.py 34.56% 53 Missing :warning:
test/test_pettingzoo.py 62.50% 3 Missing :warning:
test/test_smacv2.py 57.14% 3 Missing :warning:
test/test_vmas.py 57.14% 3 Missing :warning:
test/conftest.py 80.00% 1 Missing :warning:
Additional details and impacted files ```diff @@ Coverage Diff @@ ## main #30 +/- ## =========================================== - Coverage 89.47% 45.26% -44.21% =========================================== Files 54 56 +2 Lines 2451 2556 +105 =========================================== - Hits 2193 1157 -1036 - Misses 258 1399 +1141 ```

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

matteobettini commented 8 months ago

@janblumenkamp @acciorocketships here is the GNN!