search

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

fix: agent id obs spec dtype #1016

Closed sash-a closed 7 months ago

sash-a commented 7 months ago

What?

Agent ID wrapper assumed that the obs dtype was int (discrete) this is no longer the case as we support continuous envs.

1 liner to fix this

sash-a commented 7 months ago

Can you do the same for global_state = specs.Array(state_shape, jnp.int32, "global_state") (line 95)

Great catch @OmaymaMahjoub, fixed that one and another one in the GlobalStateWrapper