numpy>=1.17.2
torch>=1.8.1
packaging

[:python_version < "3.9"]
typing-extensions

[all]
pystoi
torchvision>=0.8
pycocotools
scipy
torch-fidelity
torchvision
lpips
pytorch-lightning>=1.5
transformers>=4.10.0
regex>=2021.9.24
nltk>=3.6
tqdm>=4.41.0

[audio]
pystoi

[detection]
torchvision>=0.8
pycocotools

[docs]
sphinx-autodoc-typehints>=1.0
nbsphinx>=0.8
docutils>=0.16
sphinx-togglebutton>=0.2
pandoc>=1.0
myst-parser
sphinx-paramlinks>=0.5.1
sphinxcontrib-fulltoc>=1.0
sphinxcontrib-mockautodoc
sphinx-copybutton>=0.3
sphinx<5.0,>=4.0

[image]
scipy
torch-fidelity
torchvision
lpips

[integrate]
pytorch-lightning>=1.5

[multimodal]
transformers>=4.10.0

[test]
types-protobuf
rouge-score>=0.0.4
bert_score==0.3.10
requests
mir_eval>=0.6
jiwer>=2.3.0
scikit-learn<1.1.1,>1.0
check-manifest
types-tabulate
pytest-timeout
types-emoji
pycocotools
coverage>5.2
pytest==6.*
types-six
kornia>=0.6.7
phmdoctest>=1.1.1
pandas
pytest-cov>2.10
cloudpickle>=1.3
pre-commit>=1.0
scipy
psutil
mypy==0.982
types-requests
pytest-rerunfailures>=10.0
types-PyYAML
types-setuptools
sacrebleu>=2.0.0
netcal
pytorch-msssim==0.2.1
transformers>4.4.0
fast-bss-eval>=0.1.0
fire
scikit-image>0.17.1
dython
torch_complex
pytest-doctestplus>=0.9.0
huggingface-hub<0.7
pypesq>1.2

[text]
regex>=2021.9.24
nltk>=3.6
tqdm>=4.41.0
