sacred>=0.8.2
torch>=1.10.0
minigrid<2.3,>=2.2
gymnasium[atari]
numpy>=1.20.3
scipy>=1.8
autorom[accept-rom-license]>=0.4.2
GPUtil>=1.4.0
procgen>=0.10.7
torchvision>=0.11.1
tensorflow-probability
tqdm
opencv-python
tensorboard

[dev]
black
flake8
pep8-naming
mypy
pytest
pytest-timeout
pytest-xdist
pytest-notebook
isort
types-psutil
ipykernel

[rllib]
ray[rllib]<2.6,>=2.5

[sb3]
stable-baselines3==2.2.1
imitation==1.0.0
