numpy>=1.17.2
torch>=1.3.1
packaging

[:python_version < "3.9"]
typing-extensions

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

[audio]
pystoi

[detection]
pycocotools
torchvision>=0.8

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

[image]
torchvision
torch-fidelity
scipy
lpips

[integrate]
pytorch-lightning>=1.5

[test]
fire
pytest-timeout
check-manifest
psutil
coverage>5.2
pytest-cov>2.10
pre-commit>=1.0
jiwer>=2.3.0
fast-bss-eval>=0.1.0
rouge-score>=0.0.4
requests
huggingface-hub<0.7
mir_eval>=0.6
pytest==6.*
transformers>=4.0
pycocotools
pypesq>1.2
sacrebleu>=2.0.0
scikit-image>0.17.1
netcal
bert_score==0.3.10
pytorch-msssim==0.2.1
phmdoctest>=1.1.1
scikit-learn<1.1.1,>1.0
pytest-doctestplus>=0.9.0
cloudpickle>=1.3
torch_complex

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