Rename to gymnasium

gymnasium/envs/box2d/bipedal_walker.py

@@ -21,7 +21,9 @@ try:
         revoluteJointDef,
     )
 except ImportError:
-    raise DependencyNotInstalled("box2D is not installed, run `pip install gymnasium[box2d]`")
+    raise DependencyNotInstalled(
+        "box2D is not installed, run `pip install gymnasium[box2d]`"
+    )
 
 
 if TYPE_CHECKING:

gymnasium/envs/box2d/car_dynamics.py

@@ -17,7 +17,9 @@ from gymnasium.error import DependencyNotInstalled
 try:
     from Box2D.b2 import fixtureDef, polygonShape, revoluteJointDef
 except ImportError:
-    raise DependencyNotInstalled("box2D is not installed, run `pip install gymnasium[box2d]`")
+    raise DependencyNotInstalled(
+        "box2D is not installed, run `pip install gymnasium[box2d]`"
+    )
 
 
 SIZE = 0.02

gymnasium/envs/box2d/car_racing.py

@@ -15,7 +15,9 @@ try:
     import Box2D
     from Box2D.b2 import contactListener, fixtureDef, polygonShape
 except ImportError:
-    raise DependencyNotInstalled("box2D is not installed, run `pip install gymnasium[box2d]`")
+    raise DependencyNotInstalled(
+        "box2D is not installed, run `pip install gymnasium[box2d]`"
+    )
 
 try:
     # As pygame is necessary for using the environment (reset and step) even without a render mode

gymnasium/envs/box2d/lunar_lander.py

@@ -23,7 +23,9 @@ try:
         revoluteJointDef,
     )
 except ImportError:
-    raise DependencyNotInstalled("box2d is not installed, run `pip install gymnasium[box2d]`")
+    raise DependencyNotInstalled(
+        "box2d is not installed, run `pip install gymnasium[box2d]`"
+    )
 
 
 if TYPE_CHECKING:

gymnasium/envs/classic_control/__init__.py

@@ -1,5 +1,7 @@
 from gymnasium.envs.classic_control.acrobot import AcrobotEnv
 from gymnasium.envs.classic_control.cartpole import CartPoleEnv
-from gymnasium.envs.classic_control.continuous_mountain_car import Continuous_MountainCarEnv
+from gymnasium.envs.classic_control.continuous_mountain_car import (
+    Continuous_MountainCarEnv,
+)
 from gymnasium.envs.classic_control.mountain_car import MountainCarEnv
 from gymnasium.envs.classic_control.pendulum import PendulumEnv

gymnasium/envs/mujoco/mujoco_env.py

@@ -395,7 +395,9 @@ class MujocoEnv(BaseMujocoEnv):
 
     def _get_viewer(
         self, mode
-    ) -> Union["gymnasium.envs.mujoco.Viewer", "gymnasium.envs.mujoco.RenderContextOffscreen"]:
+    ) -> Union[
+        "gymnasium.envs.mujoco.Viewer", "gymnasium.envs.mujoco.RenderContextOffscreen"
+    ]:
         self.viewer = self._viewers.get(mode)
         if self.viewer is None:
             if mode == "human":

gymnasium/vector/utils/__init__.py

@@ -6,8 +6,12 @@ from gymnasium.vector.utils.shared_memory import (
     read_from_shared_memory,
     write_to_shared_memory,
 )
-from gymnasium.vector.utils.spaces import _BaseGymSpaces  # pyright: reportPrivateUsage=false
+from gymnasium.vector.utils.spaces import (  # pyright: reportPrivateUsage=false
-from gymnasium.vector.utils.spaces import BaseGymSpaces, batch_space, iterate
+    BaseGymSpaces,
+    _BaseGymSpaces,
+    batch_space,
+    iterate,
+)
 
 __all__ = [
     "CloudpickleWrapper",

gymnasium/vector/utils/numpy_utils.py

@@ -5,7 +5,15 @@ from typing import Iterable, Union
 
 import numpy as np
 
-from gymnasium.spaces import Box, Dict, Discrete, MultiBinary, MultiDiscrete, Space, Tuple
+from gymnasium.spaces import (
+    Box,
+    Dict,
+    Discrete,
+    MultiBinary,
+    MultiDiscrete,
+    Space,
+    Tuple,
+)
 
 __all__ = ["concatenate", "create_empty_array"]
 

gymnasium/vector/utils/shared_memory.py

@@ -8,7 +8,15 @@ from typing import Union
 import numpy as np
 
 from gymnasium.error import CustomSpaceError
-from gymnasium.spaces import Box, Dict, Discrete, MultiBinary, MultiDiscrete, Space, Tuple
+from gymnasium.spaces import (
+    Box,
+    Dict,
+    Discrete,
+    MultiBinary,
+    MultiDiscrete,
+    Space,
+    Tuple,
+)
 
 __all__ = ["create_shared_memory", "read_from_shared_memory", "write_to_shared_memory"]
 

gymnasium/vector/utils/spaces.py

@@ -7,7 +7,15 @@ from typing import Iterator
 import numpy as np
 
 from gymnasium.error import CustomSpaceError
-from gymnasium.spaces import Box, Dict, Discrete, MultiBinary, MultiDiscrete, Space, Tuple
+from gymnasium.spaces import (
+    Box,
+    Dict,
+    Discrete,
+    MultiBinary,
+    MultiDiscrete,
+    Space,
+    Tuple,
+)
 
 BaseGymSpaces = (Box, Discrete, MultiDiscrete, MultiBinary)
 _BaseGymSpaces = BaseGymSpaces

gymnasium/wrappers/compatibility.py

@@ -4,7 +4,9 @@ from typing import Any, Dict, Optional, Tuple
 
 import gymnasium
 from gymnasium.core import ObsType
-from gymnasium.utils.step_api_compatibility import convert_to_terminated_truncated_step_api
+from gymnasium.utils.step_api_compatibility import (
+    convert_to_terminated_truncated_step_api,
+)
 
 if sys.version_info >= (3, 8):
     from typing import Protocol, runtime_checkable

gymnasium/wrappers/order_enforcing.py

@@ -19,7 +19,9 @@ class OrderEnforcing(gymnasium.Wrapper):
         >>> env.step(0)
     """
 
-    def __init__(self, env: gymnasium.Env, disable_render_order_enforcing: bool = False):
+    def __init__(
+        self, env: gymnasium.Env, disable_render_order_enforcing: bool = False
+    ):
         """A wrapper that will produce an error if :meth:`step` is called before an initial :meth:`reset`.
 
         Args:

gymnasium/wrappers/render_collection.py

@@ -5,7 +5,9 @@ import gymnasium
 class RenderCollection(gymnasium.Wrapper):
     """Save collection of render frames."""
 
-    def __init__(self, env: gymnasium.Env, pop_frames: bool = True, reset_clean: bool = True):
+    def __init__(
+        self, env: gymnasium.Env, pop_frames: bool = True, reset_clean: bool = True
+    ):
         """Initialize a :class:`RenderCollection` instance.
 
         Args:

setup.py

@@ -72,7 +72,9 @@ setup(
     long_description=long_description,
     long_description_content_type="text/markdown",
     name="gymnasium",
-    packages=[package for package in find_packages() if package.startswith("gymnasium")],
+    packages=[
+        package for package in find_packages() if package.startswith("gymnasium")
+    ],
     package_data={
         "gymnasium": [
             "envs/mujoco/assets/*.xml",

tests/envs/test_env_implementation.py

@@ -72,7 +72,9 @@ def test_bipedal_walker_hardcore_creation(seed: int):
     HC_TERRAINS_COLOR2 = (153, 153, 153)
 
     env = gymnasium.make("BipedalWalker-v3", disable_env_checker=True).unwrapped
-    hc_env = gymnasium.make("BipedalWalkerHardcore-v3", disable_env_checker=True).unwrapped
+    hc_env = gymnasium.make(
+        "BipedalWalkerHardcore-v3", disable_env_checker=True
+    ).unwrapped
     assert isinstance(env, BipedalWalker) and isinstance(hc_env, BipedalWalker)
     assert env.hardcore is False and hc_env.hardcore is True
 

tests/envs/test_make.py

@@ -9,7 +9,12 @@ import pytest
 
 import gymnasium
 from gymnasium.envs.classic_control import cartpole
-from gymnasium.wrappers import AutoResetWrapper, HumanRendering, OrderEnforcing, TimeLimit
+from gymnasium.wrappers import (
+    AutoResetWrapper,
+    HumanRendering,
+    OrderEnforcing,
+    TimeLimit,
+)
 from gymnasium.wrappers.env_checker import PassiveEnvChecker
 from tests.envs.test_envs import PASSIVE_CHECK_IGNORE_WARNING
 from tests.envs.utils import all_testing_env_specs
@@ -69,7 +74,8 @@ def test_make_max_episode_steps():
     env = gymnasium.make("CartPole-v1", disable_env_checker=True)
     assert has_wrapper(env, TimeLimit)
     assert (
-            env.spec.max_episode_steps == gymnasium.envs.registry["CartPole-v1"].max_episode_steps
+        env.spec.max_episode_steps
+        == gymnasium.envs.registry["CartPole-v1"].max_episode_steps
     )
     env.close()
 

tests/envs/test_register.py

@@ -86,7 +86,9 @@ def test_register(
     ],
 )
 def test_register_error(env_id):
-    with pytest.raises(gymnasium.error.Error, match=f"^Malformed environment ID: {env_id}"):
+    with pytest.raises(
+        gymnasium.error.Error, match=f"^Malformed environment ID: {env_id}"
+    ):
         gymnasium.register(env_id, "no-entry-point")
 
 

tests/spaces/test_tuple.py

@@ -89,19 +89,25 @@ def test_bad_space_calls(space_fn):
 
 
 def test_contains_promotion():
-    space = gymnasium.spaces.Tuple((gymnasium.spaces.Box(0, 1), gymnasium.spaces.Box(-1, 0, (2,))))
+    space = gymnasium.spaces.Tuple(
+        (gymnasium.spaces.Box(0, 1), gymnasium.spaces.Box(-1, 0, (2,)))
+    )
 
     assert (
         np.array([0.0], dtype=np.float32),
         np.array([0.0, 0.0], dtype=np.float32),
     ) in space
 
-    space = gymnasium.spaces.Tuple((gymnasium.spaces.Box(0, 1), gymnasium.spaces.Box(-1, 0, (1,))))
+    space = gymnasium.spaces.Tuple(
+        (gymnasium.spaces.Box(0, 1), gymnasium.spaces.Box(-1, 0, (1,)))
+    )
     assert np.array([[0.0], [0.0]], dtype=np.float32) in space
 
 
 def test_bad_seed():
-    space = gymnasium.spaces.Tuple((gymnasium.spaces.Box(0, 1), gymnasium.spaces.Box(0, 1)))
+    space = gymnasium.spaces.Tuple(
+        (gymnasium.spaces.Box(0, 1), gymnasium.spaces.Box(0, 1))
+    )
     with pytest.raises(
         TypeError,
         match="Expected seed type: list, tuple, int or None, actual type: <class 'float'>",

tests/spaces/test_utils.py

@@ -92,7 +92,8 @@ def test_flatten_space(space):
             assert edge_single_dim == edge_flatdim
     else:
         assert isinstance(
-            space, (gymnasium.spaces.Tuple, gymnasium.spaces.Dict, gymnasium.spaces.Sequence)
+            space,
+            (gymnasium.spaces.Tuple, gymnasium.spaces.Dict, gymnasium.spaces.Sequence),
         )
 
 

tests/utils/test_play.py

@@ -188,7 +188,9 @@ def test_play_loop_real_env():
             action = keys_to_action[chr(e.key) if str_keys else (e.key,)]
             obs, _, _, _, _ = env.step(action)
 
-        env_play = gymnasium.make(ENV, render_mode="rgb_array", disable_env_checker=True)
+        env_play = gymnasium.make(
+            ENV, render_mode="rgb_array", disable_env_checker=True
+        )
         if apply_wrapper:
             env_play = KeysToActionWrapper(env, keys_to_action=keys_to_action)
             assert hasattr(env_play, "get_keys_to_action")

tests/vector/test_async_vector_env.py

@@ -4,7 +4,11 @@ from multiprocessing import TimeoutError
 import numpy as np
 import pytest
 
-from gymnasium.error import AlreadyPendingCallError, ClosedEnvironmentError, NoAsyncCallError
+from gymnasium.error import (
+    AlreadyPendingCallError,
+    ClosedEnvironmentError,
+    NoAsyncCallError,
+)
 from gymnasium.spaces import Box, Discrete, MultiDiscrete, Tuple
 from gymnasium.vector.async_vector_env import AsyncVectorEnv
 from tests.vector.utils import (

tests/wrappers/test_record_video.py

@@ -26,8 +26,12 @@ def test_record_video_using_default_trigger():
 
 
 def test_record_video_reset():
-    env = gymnasium.make("CartPole-v1", render_mode="rgb_array", disable_env_checker=True)
+    env = gymnasium.make(
-    env = gymnasium.wrappers.RecordVideo(env, "videos", step_trigger=lambda x: x % 100 == 0)
+        "CartPole-v1", render_mode="rgb_array", disable_env_checker=True
+    )
+    env = gymnasium.wrappers.RecordVideo(
+        env, "videos", step_trigger=lambda x: x % 100 == 0
+    )
     ob_space = env.observation_space
     obs, info = env.reset()
     env.close()
@@ -38,9 +42,13 @@ def test_record_video_reset():
 
 
 def test_record_video_step_trigger():
-    env = gymnasium.make("CartPole-v1", render_mode="rgb_array", disable_env_checker=True)
+    env = gymnasium.make(
+        "CartPole-v1", render_mode="rgb_array", disable_env_checker=True
+    )
     env._max_episode_steps = 20
-    env = gymnasium.wrappers.RecordVideo(env, "videos", step_trigger=lambda x: x % 100 == 0)
+    env = gymnasium.wrappers.RecordVideo(
+        env, "videos", step_trigger=lambda x: x % 100 == 0
+    )
     env.reset()
     for _ in range(199):
         action = env.action_space.sample()

tests/wrappers/test_transform_observation.py

@@ -12,7 +12,8 @@ def test_transform_observation(env_id):
 
     env = gymnasium.make(env_id, disable_env_checker=True)
     wrapped_env = TransformObservation(
-        gymnasium.make(env_id, disable_env_checker=True), lambda obs: affine_transform(obs)
+        gymnasium.make(env_id, disable_env_checker=True),
+        lambda obs: affine_transform(obs),
     )
 
     obs, info = env.reset(seed=0)

tests/wrappers/test_transform_reward.py

@@ -30,7 +30,8 @@ def test_transform_reward(env_id):
     max_r = 0.0002
     env = gymnasium.make(env_id, disable_env_checker=True)
     wrapped_env = TransformReward(
-        gymnasium.make(env_id, disable_env_checker=True), lambda r: np.clip(r, min_r, max_r)
+        gymnasium.make(env_id, disable_env_checker=True),
+        lambda r: np.clip(r, min_r, max_r),
     )
     action = env.action_space.sample()
 

tests/wrappers/test_vector_list_info.py

@@ -11,7 +11,9 @@ SEED = 42
 
 def test_usage_in_vector_env():
     env = gymnasium.make(ENV_ID, disable_env_checker=True)
-    vector_env = gymnasium.vector.make(ENV_ID, num_envs=NUM_ENVS, disable_env_checker=True)
+    vector_env = gymnasium.vector.make(
+        ENV_ID, num_envs=NUM_ENVS, disable_env_checker=True
+    )
 
     VectorListInfo(vector_env)
 
@@ -20,7 +22,9 @@ def test_usage_in_vector_env():
 
 
 def test_info_to_list():
-    env_to_wrap = gymnasium.vector.make(ENV_ID, num_envs=NUM_ENVS, disable_env_checker=True)
+    env_to_wrap = gymnasium.vector.make(
+        ENV_ID, num_envs=NUM_ENVS, disable_env_checker=True
+    )
     wrapped_env = VectorListInfo(env_to_wrap)
     wrapped_env.action_space.seed(SEED)
     _, info = wrapped_env.reset(seed=SEED)
@@ -38,7 +42,9 @@ def test_info_to_list():
 
 
 def test_info_to_list_statistics():
-    env_to_wrap = gymnasium.vector.make(ENV_ID, num_envs=NUM_ENVS, disable_env_checker=True)
+    env_to_wrap = gymnasium.vector.make(
+        ENV_ID, num_envs=NUM_ENVS, disable_env_checker=True
+    )
     wrapped_env = VectorListInfo(RecordEpisodeStatistics(env_to_wrap))
     _, info = wrapped_env.reset(seed=SEED)
     wrapped_env.action_space.seed(SEED)