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: add time_limit to jumanji env config #991

Closed WiemKhlifi closed 8 months ago

WiemKhlifi commented 8 months ago

What?

Add the time_limit attribute to Jumanji's environments.

Why?

This helps control the time_limit and avoid using the default one.

How?

Extract this attribute from config and feed it to the make function as jumanji.make(...., **config.env.kwargs).