Fix: dtype of agentid wrapper.

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

Closed lbeyers closed 7 months ago

lbeyers commented 7 months ago

Fixes #1032 :)

Dtype was not enforced to be consistent, caused errors.

Enforced dtype in concatenation operation.

CLAassistant commented 7 months ago

All committers have signed the CLA.