search

instadeepai / jumanji

🕹️ A diverse suite of scalable reinforcement learning environments in JAX
https://instadeepai.github.io/jumanji
Apache License 2.0
583 stars 70 forks source link

fix: flatpack was training with ints #234

Closed sash-a closed 5 months ago

sash-a commented 5 months ago

flatpack training was returning an error: ValueError: data type <class 'numpy.int32'> not inexact because we changed the dtypes of the observation to ints, I have casted it to floats in the training.

Thanks @RuanJohn for reporting this :smile: