# Deep learning.
# --------------
tensorflow==2.4.1
tensorflow-probability==0.12.2
torch==1.8.1;sys_platform=="darwin"
torchvision==0.9.1;sys_platform=="darwin"

# On non-OSX machines only install CPU version of torch and torchvision
-f https://download.pytorch.org/whl/torch_stable.html
torch==1.8.1+cpu;sys_platform!="darwin"
-f https://download.pytorch.org/whl/torch_stable.html
torchvision==0.9.1+cpu;sys_platform!="darwin"

# Environment adapters.
# ---------------------
# Atari
atari_py==0.2.6
gym[atari]==0.18.0
# Kaggle envs.
kaggle_environments==1.7.11
# Unity3D testing
mlagents==0.26.0
mlagents_envs==0.26.0
# For tests on PettingZoo's multi-agent envs.
pettingzoo==1.8.1
# For testing in MuJoCo-like envs (in PyBullet).
pybullet==3.1.7
# For tests on RecSim and Kaggle envs.
recsim==0.2.4

# Other.
# ------
# For MAML on PyTorch.
higher==0.2.1
# For auto-generating an env-rendering Window.
pyglet==1.5.0
# For JSON reader/writer.
smart_open==5.0.0
# Ray Serve example
starlette==0.14.2