fix: non-jittable code for minatar freeway and asterix

RobertTLange / gymnax

RL Environments in JAX 🌍

Apache License 2.0

577 stars 54 forks source link

fix: non-jittable code for minatar freeway and asterix #74

Closed EdanToledo closed 2 months ago

EdanToledo commented 2 months ago

Hello,

I'm not sure how this happened or if for some reason its intentional but freeway and asterix cannot be compiled due to two really small things (at least for me)

RobertTLange commented 2 months ago

Thank you so much! There were some major type check changes and this must have broken things. The fix is merged.

P.S.: Super cool work on stoicx!