search

RobertTLange / gymnax

RL Environments in JAX 🌍
Apache License 2.0
613 stars 61 forks source link

Adding Wrappers to Gymnax #49

Closed luchris429 closed 1 year ago

luchris429 commented 1 year ago

Adds a GymnaxWrapper base class with two example wrappers. One flattens the observation, the other adds logging information.

Not sure what black settings you use!