fix: default value for obs in extras

instadeepai / jumanji

🕹️ A diverse suite of scalable reinforcement learning environments in JAX

https://instadeepai.github.io/jumanji

Apache License 2.0

645 stars 80 forks source link

Closed clement-bonnet closed 8 months ago

clement-bonnet commented 8 months ago

give a default behavior for next_obs_in_extras that is backward compatible for AutoResetWrapper and VmapAutoResetWrapper, following #223
handle non-numeric data in loggers (was crashing for next_obs of type Observation for instance)
also made the add_obs_to_extras function take a timestep instead of (state, timestep)