[Feature] Enable support for GNNs from `torch_geometric.nn.attention`

[Xmaster6y]

Allows other GNN types from torch_geometric.nn.attention.

[codecov-commenter]

Codecov Report

:x: Patch coverage is 63.63636% with 4 lines in your changes missing coverage. Please review. :white_check_mark: Project coverage is 78.13%. Comparing base (cbe5c58) to head (a2310c5). :warning: Report is 1 commits behind head on main.

Files with missing lines	Patch %	Lines
benchmarl/models/gnn.py	63.63%	4 Missing :warning:

Additional details and impacted files

@@            Coverage Diff             @@
##             main     #227      +/-   ##
==========================================
- Coverage   78.55%   78.13%   -0.42%     
==========================================
  Files          85       85              
  Lines        4448     4451       +3     
==========================================
- Hits         3494     3478      -16     
- Misses        954      973      +19     

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

:rocket: New features to boost your workflow:

• :snowflake: Test Analytics: Detect flaky tests, report on failures, and find test suite problems.