[BUG] checkpointing saving using timestep can break due to not being an int
EdanToledo opened this issue · comments
Describe the bug
Title says it all
instadeepai / Mava
🦁 A research-friendly codebase for fast experimentation of multi-agent reinforcement learning in JAX