gaspersic/Gymnasium
1
0
Fork 0
mirror of https://github.com/Farama-Foundation/Gymnasium.git synced 2025-07-31 22:04:31 +00:00
Code Issues Packages Projects Releases Wiki Activity

Update jax environment names (#414)

Browse Source
This commit is contained in:
Mark Towers
2023-03-30 14:00:14 +01:00
committed by GitHub
parent 4577717bc9
commit 44bdfa1fe0
4 changed files with 9 additions and 9 deletions
Download Patch File Download Diff File Expand all files Collapse all files

6
docs/scripts/gen_mds.py
View File

@@ -14,9 +14,9 @@ exclude_env_names = [
"FrozenLake8x8",
"LunarLanderContinuous",
"BipedalWalkerHardcore",
"CartPoleJax",
"PendulumJax",
"Jax-Blackjack",
"phys2d/CartPole",
"phys2d/Pendulum",
"tabular/Blackjack",
]
for env_spec in gym.registry.values():
if env_spec.name not in exclude_env_names:

8
gymnasium/envs/__init__.py
View File

@@ -62,7 +62,7 @@ register(
# ----------------------------------------
register(
id="CartPoleJax-v0",
id="phys2d/CartPole-v0",
entry_point="gymnasium.envs.phys2d.cartpole:CartPoleJaxEnv",
vector_entry_point="gymnasium.envs.phys2d.cartpole:CartPoleJaxVectorEnv",
max_episode_steps=200,
@@ -70,7 +70,7 @@ register(
)
register(
id="CartPoleJax-v1",
id="phys2d/CartPole-v1",
entry_point="gymnasium.envs.phys2d.cartpole:CartPoleJaxEnv",
vector_entry_point="gymnasium.envs.phys2d.cartpole:CartPoleJaxVectorEnv",
max_episode_steps=500,
@@ -78,7 +78,7 @@ register(
)
register(
id="PendulumJax-v0",
id="phys2d/Pendulum-v0",
entry_point="gymnasium.envs.phys2d.pendulum:PendulumJaxEnv",
vector_entry_point="gymnasium.envs.phys2d.pendulum:PendulumJaxVectorEnv",
max_episode_steps=200,
@@ -166,7 +166,7 @@ register(
# ----------------------------------------
register(
id="Jax-Blackjack-v0",
id="tabular/Blackjack-v0",
entry_point="gymnasium.envs.tabular.blackjack:BlackJackJaxEnv",
kwargs={"sutton_and_barto": True, "natural": False},
)

2
gymnasium/experimental/wrappers/rendering.py
View File

@@ -317,7 +317,7 @@ class HumanRenderingV0(
``render_mode="human"`` to ``make``. The wrapper will only be applied if the environment does not
implement human-rendering natively (i.e. ``render_mode`` does not contain ``"human"``).
>>> env = gym.make("CartPoleJax-v1", render_mode="human") # CartPoleJax-v1 doesn't implement human-rendering natively
>>> env = gym.make("phys2d/CartPole-v1", render_mode="human") # CartPoleJax-v1 doesn't implement human-rendering natively
>>> obs, _ = env.reset() # This will start rendering to the screen
Warning: If the base environment uses ``render_mode="rgb_array_list"``, its (i.e. the *base environment's*) render method

2
gymnasium/wrappers/human_rendering.py
View File

@@ -28,7 +28,7 @@ class HumanRendering(gym.Wrapper, gym.utils.RecordConstructorArgs):
``render_mode="human"`` to ``make``. The wrapper will only be applied if the environment does not
implement human-rendering natively (i.e. ``render_mode`` does not contain ``"human"``).
>>> env = gym.make("CartPoleJax-v1", render_mode="human") # CartPoleJax-v1 doesn't implement human-rendering natively
>>> env = gym.make("phys2d/CartPole-v1", render_mode="human") # phys2d/CartPole-v1 doesn't implement human-rendering natively
>>> obs, _ = env.reset() # This will start rendering to the screen
Warning: If the base environment uses ``render_mode="rgb_array_list"``, its (i.e. the *base environment's*) render method