numpy>=1.17.2
torch>=1.8.1
packaging

[:python_version < "3.9"]
typing-extensions

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

[audio]
pystoi

[detection]
pycocotools
torchvision>=0.8

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

[image]
lpips
torchvision
torch-fidelity
scipy

[integrate]
pytorch-lightning>=1.5

[multimodal]
transformers>=4.10.0

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

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