# Tests
numpy
pytest
codecov
pytest-cov
pytest-xdist
dill
# workaround for issue fixed in https://github.com/pytorch/pytorch/pull/69904
# to be removed after next Pytroch release (1.11)
setuptools==59.5.0
# 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
