numpy>=1.17.2
torch>=1.3.1
packaging

[:python_version < "3.9"]
typing-extensions

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

[audio]
pystoi

[detection]
pycocotools
torchvision>=0.8

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

[image]
torch-fidelity
torchvision
scipy
lpips

[integrate]
pytorch-lightning>=1.5

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

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