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: cast timestep to int #1010

Closed callumtilbury closed 7 months ago

callumtilbury commented 7 months ago

What?

Cast timestep to int before logging.

Why?

Fixes #1009.

How?

Cast to int.

Extra

callumtilbury commented 7 months ago

@EdanToledo can you check that this is ok? Thanks pal!