RobertTLange / gymnax

RL Environments in JAX 🌍

Geek Repo:Geek Repo

Github PK Tool:Github PK Tool

Gymnasium API update

smorad opened this issue · comments

Will gymnax update their API in-line with gymnasium? Namely the recent breaking change in gym/gymnasium where step returns terminated, truncated instead of done?

+1

The use of gym in this package is optional, and it could be replaced by import gymnasium as gym and as a fallback to gym. The package itself should not depend on gym.

Hi,

I would like to bring this subject back up as it is important.

Is there any plan to switch the current gymnax API (that returns done) to the gymnasium API (the returns terminatedand truncated) ?

I know that gymnax doesn't rely on gym to function, I'm talking about the return of the gymnax environments themselves.

As to why it's important to allow for the distinction between terminatedand truncated, here is a nice explication from gymnasium : https://farama.org/Gymnasium-Terminated-Truncated-Step-API.

If you'd prefer I can write the PR myself.