shimmy_legacy_gym_to_gymnasium_compatibility_wrapper.py

1import gymnasium as gym
2import shimmy
3
4# Load a legacy gym environment using the shimmy compatibility layer
5# Note: This requires the specific legacy environment to be installed (e.g., pip install gym[atari])
6env = gym.make("GymV26Environment-v0", env_id="CartPole-v1")
7
8# Standard Gymnasium API loop
9observation, info = env.reset()
10
11for _ in range(1000):
12    action = env.action_space.sample()  # agent policy that takes a random action
13    observation, reward, terminated, truncated, info = env.step(action)
14
15    if terminated or truncated:
16        observation, info = env.reset()
17
18env.close()