[Refactor] Pass experiment object to algorithm

facebookresearch / BenchMARL

A collection of MARL benchmarks based on TorchRL

https://benchmarl.readthedocs.io/

MIT License

216 stars 27 forks source link

[Refactor] Pass experiment object to algorithm #31

Closed matteobettini closed 10 months ago

matteobettini commented 10 months ago

This PR makes it so that the whole experiment object is passed to algorithms in order to increment the available inoformation to a user implementing a custom algoritim.

Furtermore it:

makes the scale transformation in probabilistic algorithms customizable by the user
allow models to access the action spec