setuptools>=67.0
gymnasium
click
termcolor
gym
torch
treevalue
rich
wandb
seaborn
jsonargparse
imageio
opencv-python
pygame

[dev]
build
twine

[mpe]
pyglet==1.5.27

[nlp]
transformers==4.18.0
datasets
nltk
evaluate
icetk

[retro]
gym-retro

[super_mario]
gym-super-mario-bros

[test]
pytest
pytest-cov
mypy
isort
black
ruff
gpustat
gym-super-mario-bros
