instadeepai / Mava

🦁 A research-friendly codebase for fast experimentation of multi-agent reinforcement learning in JAX
Apache License 2.0
709 stars 83 forks source link

Feat: Jumanji register manager #980

Closed WiemKhlifi closed 8 months ago

WiemKhlifi commented 8 months ago

What?

Implement a registry manager class for Jumanji environments. In this case, we check the existence of the given scenario and create the corresponding attributes to initialise the environment class.

Why?

This manager class and the accompanied functions help avoid creating numerous configuration files for each scenario (e.g. for RWARE and LBF).

How?

For the selected environment-scenario pair:

Extra

Teaser: This also helps merge the gym and Jumanji configs for the two common environments used (this will be the focus of a separate PR).

Close #975.

WiemKhlifi commented 8 months ago

The problem is that for each Jumanji environment we add, we'll have to add a specific function to extract its attributes, which will increase the number of functions added. So we've decided to close this PR for now and use YAML configurations and treat this as an environment specification that hopefully can be handled in Jumanji itself in the future.

Changes will remain on the feat/jumanji_env_register branch for a while 🤞