Giters

instadeepai / Mava

🦁 A research-friendly codebase for fast experimentation of multi-agent reinforcement learning in JAX

Home Page:https://id-mava.readthedocs.io

Geek Repo:Geek Repo

Github PK Tool:Github PK Tool

[MAINTAIN] Better equality check for jaxmarl specs

sash-a opened this issue · comments

commented

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.