search

RobertTLange / gymnax

RL Environments in JAX 🌍
Apache License 2.0
577 stars 54 forks source link

Fix reward dtype in FourRooms #60

Open jiriyu98 opened 11 months ago

jiriyu98 commented 11 months ago

Based on type annotation and according to tradition, I think this behavior is unexpected.

jnp.logical_and won't convert reward dtype from bool to float. Simply add *1.0 to convert it.