requirements-dev.txt

# Tests
numpy
pytest
pytest-cov
pytest-xdist
pytest-timeout
dill
filelock
setuptools
# Test contrib dependencies
scipy
pytorch_fid
tqdm
scikit-learn
matplotlib
tensorboardX
visdom
polyaxon
wandb
mlflow
neptune-client>=0.16.17
tensorboard
torchvision
pynvml<12  # pynvml module was removed in 12.X, is not developed or maintained. We should replace pynvml with something else.
clearml
scikit-image
py-rouge
pycocotools
# temporary fix for python=3.12 and v3.8.1
# nltk
git+https://github.com/nltk/nltk@aba99c8
# Examples dependencies
pandas
gymnasium
# temporary fix: E   AttributeError: module 'mpmath' has no attribute 'rational'
mpmath<1.4