dm-acme~=0.2.4
chex
absl-py
dm_env
dm-tree
numpy~=1.21.4
pillow
matplotlib
dataclasses
box2d-py
gym

[envs]
pettingzoo~=1.13.1
multi_agent_ale_py
supersuit==3.3.1
pygame
pysc2
open_spiel
pysc2
SMAC@ git+https://github.com/oxwhirl/smac.git

[flatland]
flatland-rl==3.0.1

[launchpad]
dm-launchpad~=0.3.2

[open_spiel]
open_spiel

[pz]
pettingzoo~=1.13.1
multi_agent_ale_py
supersuit==3.3.1
pygame
pysc2

[record_episode]
array2gif

[reverb]
dm-reverb~=0.6.1
jax
jaxlib

[sc2]
pysc2
SMAC@ git+https://github.com/oxwhirl/smac.git

[testing_formatting]
pytest==6.2.4
pre-commit
mypy==0.910
pytest-xdist
flake8==3.8.2
black==21.4b1
pytest-cov
interrogate
pydocstyle
types-six

[tf]
tensorflow~=2.7.0
tensorflow_probability~=0.15.0
dm-sonnet
trfl
