forked from instadeepai/Mava
-
Notifications
You must be signed in to change notification settings - Fork 0
/
Copy pathrequirements.txt
75 lines (63 loc) · 1016 Bytes
/
requirements.txt
1
2
3
4
5
6
7
8
9
10
11
12
13
14
15
16
17
18
19
20
21
22
23
24
25
26
27
28
29
30
31
32
33
34
35
36
37
38
39
40
41
42
43
44
45
46
47
48
49
50
51
52
53
54
55
56
57
58
59
60
61
62
63
64
65
66
67
68
69
70
71
72
73
74
# Refer to https://github.com/instadeepai/Mava/blob/develop/setup.py
# Miscell
absl-py
dm_env
dm-tree
numpy==1.22.4
pillow
matplotlib
dataclasses
# Run this 1st: sudo apt install swig
box2d-py
gym<=0.23.0
# gym==0.14.0
importlib-metadata<5.0
typing-extensions
cloudpickle==2.2.1
reverb==2.0.1
# Tensorflow Requirements
tensorflow==2.8.0
# tensorflow==2.12.0
tensorflow_probability==0.16.0
# tensorflow_probability==0.15.0
dm-sonnet
dm-launchpad
dm-acme==0.4.0
swig
trfl
chex
dm-reverb==0.7.3
tf-agents[reverb]==0.12.0
# Jax Requirements
# jax==0.3.24
# jaxlib==0.3.24
jax==0.3.15
jaxlib==0.3.15
dm-haiku==0.0.8
flax
optax==0.1.4
rlax
networkx
# Testing Formatting Requirements
# pytest==7.2.0
pytest==4.6.11
pre-commit
mypy==0.981
pytest-xdist
flake8==3.8.2
black==22.3.0
pytest-cov
interrogate
pydocstyle
types-six
# Record episode requirements
array2gif
pyglet
# Flatland_requirements
# flatland-rl==3.0.1
# PettingZoo Requirements
pettingzoo==1.17.0
multi_agent_ale_py
supersuit==3.3.4
pygame
pysc2