instadeepai / Mava

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

[MAINTAIN] Better equality check for jaxmarl specs #961

Open sash-a opened 11 months ago

sash-a commented 11 months ago

Please describe what needs to be maintained?

This came up in #955. Unfortunately it's quite messy to check the equality of two jaxmarl spaces as they don't have custom __eq__ methods. So the current solution is to check that only the shapes match. This could cause possible bugs in the future so it would be great to find a better solution.