# Tests
numpy
pytest
codecov
pytest-cov
pytest-xdist
dill==0.3.4
setuptools
# Test contrib dependencies
scipy
pytorch_fid==0.1.1
tqdm
scikit-learn
matplotlib
tensorboardX
visdom
# temporary fix for
# ImportError: cannot import name 'soft_unicode' from 'markupsafe'
markupsafe==2.0.1
polyaxon
polyaxon-client
wandb
mlflow
neptune-client
tensorboard
torchvision
pynvml
clearml
scikit-image
py-rouge
nltk
# Examples dependencies
pandas
gym
# pygame is now required by gym for cartpole env
pygame
mkl
