Fix/env equivalence vs popgym#1

Open

noahfarr wants to merge 14 commits into

FLAIROx:mainfrom

noahfarr:fix/env-equivalence-vs-popgym

Commits on Jul 7, 2026

fix: repair broken gymnax.wrappers.purerl import, declare chex dependency
noahfarr
committed
feat: auto-wrap envs that declare obs_requires_prev_action
noahfarr
committed
fix: Autoencode observation inversion, OOB read, off-by-one episode length
noahfarr
committed
fix: Battleship ship placement bias, missing prev-action contract
noahfarr
committed
fix: Concentration episode_length dtype, missing prev-action contract
noahfarr
committed
fix: CountRecall hardcoded action space, timing, reset, reward scale, length
noahfarr
committed
fix: HigherLower off-by-one episode length causing OOB terminal read
noahfarr
committed
fix: MineSweeper undersized obs, unreachable win, mine overwrite, timing
noahfarr
committed
fix: MultiArmedBandit unlearnable without prev-action, empty info dict
noahfarr
committed
fix: RepeatFirst off-by-one episode length leaking the target card
noahfarr
committed
fix: RepeatPrevious recall lag off by one, off-by-one episode length
noahfarr
committed
fix: CartPole spurious post-termination reward, unconditional noise clipping
noahfarr
committed
fix: Pendulum noisy observation not clipped to bounds
noahfarr
committed
fix: meta_cartpole compat with newer JAX (jax.tree_map removed)
noahfarr
committed