An API conversion tool for reinforcement learning environments.
Shimmy provides Gymnasium and PettingZoo bindings for popular external RL environments.
Environments can be interacted with using a simple, high-level API:
import gymnasium as gym
env = gym.make("dm_control/acrobot-swingup_sparse-v0", render_mode="human")
observation, info = env.reset(seed=42)
for _ in range(1000):
action = env.action_space.sample() # this is where you would insert your policy
observation, reward, terminated, truncated, info = env.step(action)
if terminated or truncated:
observation, info = env.reset()
env.close()