corl-team / xland-minigrid

JAX-accelerated Meta-Reinforcement Learning Environments Inspired by XLand and MiniGrid 🏎️
Apache License 2.0
162 stars 12 forks source link

implement jit-compatible rendering #8

Closed Howuhh closed 3 months ago

Howuhh commented 4 months ago

to provide image observations as alternative to symbolic encoding

Howuhh commented 4 months ago

To make it easier, I think we should unify END_OF_MAP, UNSEEN and EMPTY tiles, mapping them to just EMPTY. However, this will shift all tiles ids breaking benchmarks. Right now I don't want to re-generate them again, so I'll put that on hold for now. Let's do it!