torch
numpy
gymnasium
stable-baselines3
tqdm
wandb
imageio[ffmpeg]
opencv-python
wandb[media]
pygame-ce>=2.1.3

[all]
rich
psutil
nvidia-ml-py
ale-py
swig
gymnasium[atari,box2d,classic-control,mujoco]

[atari]
gymnasium[atari]
ale-py

[box2d]
swig
gymnasium[box2d]

[classic]
gymnasium[classic-control]

[cli]
rich
psutil
nvidia-ml-py

[dev]
ruff
mypy
black
build
twine
rich
psutil
nvidia-ml-py

[mujoco]
gymnasium[mujoco]
