search

facebookresearch / BenchMARL

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

[BugFix] GNN position and velocity key #132

Closed matteobettini closed 2 months ago

matteobettini commented 2 months ago

This PR amends #125 which suggested to include position key and velocity key in gnns in the info dict.

This solution is not feasible as torchrl impedes access to this information in some losses.

The current supported way to include these keys in as part of the observations.

This will however require the GNN to be the first in the sequence if SequenceModel is used