numpy>1.20.0
torch>=1.8.1
packaging
lightning-utilities>=0.7.0

[:python_version < "3.9"]
typing-extensions

[all]
pystoi<=0.3.3
pycocotools>2.0.0
torchvision>=0.8
torch-fidelity<=0.3.0
scipy>1.0.0
lpips<=0.1.4
transformers>=4.10.0
nltk>=3.6
tqdm>=4.41.0
regex>=2021.9.24
mypy==1.4.1
types-six
types-emoji
types-requests
types-setuptools
types-protobuf
types-tabulate
types-PyYAML
matplotlib>=3.2.0
SciencePlots>=2.0.0

[audio]
pystoi<=0.3.3

[detection]
pycocotools>2.0.0
torchvision>=0.8

[dev]
pystoi<=0.3.3
pycocotools>2.0.0
torchvision>=0.8
torch-fidelity<=0.3.0
scipy>1.0.0
lpips<=0.1.4
transformers>=4.10.0
nltk>=3.6
tqdm>=4.41.0
regex>=2021.9.24
mypy==1.4.1
types-six
types-emoji
types-requests
types-setuptools
types-protobuf
types-tabulate
types-PyYAML
matplotlib>=3.2.0
SciencePlots>=2.0.0
huggingface-hub<0.16
coverage==7.2.7
scikit-learn>=1.1.1
pytest-timeout==2.1.0
scikit-image>=0.19.0
requests<=2.31.0
cloudpickle>1.3
dython<=0.7.4
sacrebleu>=2.0.0
pytest-cov==4.1.0
pandas>=1.4.0
pytorch-msssim==1.0.0
bert_score==0.3.13
kornia>=0.6.7
mir-eval>=0.6
sewar>=0.4.4
rouge-score>0.1.0
psutil<=5.9.5
statsmodels>0.13.5
torch_complex<=0.4.3
pytest-doctestplus==0.13.0
pandas>1.0.0
fire<=0.5.0
fast-bss-eval>=0.1.0
transformers>4.4.0
phmdoctest==1.4.0
fairlearn
pytest==7.4.0
netcal>1.0.0
jiwer>=2.3.0
pytest-rerunfailures==11.1.2
numpy<1.25.0

[image]
torchvision>=0.8
torch-fidelity<=0.3.0
scipy>1.0.0
lpips<=0.1.4

[multimodal]
transformers>=4.10.0

[test]
huggingface-hub<0.16
coverage==7.2.7
scikit-learn>=1.1.1
pytest-timeout==2.1.0
scikit-image>=0.19.0
requests<=2.31.0
cloudpickle>1.3
dython<=0.7.4
sacrebleu>=2.0.0
pytest-cov==4.1.0
pandas>=1.4.0
pytorch-msssim==1.0.0
scipy>1.0.0
bert_score==0.3.13
kornia>=0.6.7
mir-eval>=0.6
sewar>=0.4.4
rouge-score>0.1.0
psutil<=5.9.5
statsmodels>0.13.5
torch_complex<=0.4.3
pytest-doctestplus==0.13.0
pandas>1.0.0
fire<=0.5.0
fast-bss-eval>=0.1.0
transformers>4.4.0
phmdoctest==1.4.0
fairlearn
pytest==7.4.0
netcal>1.0.0
jiwer>=2.3.0
pytest-rerunfailures==11.1.2
numpy<1.25.0

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

[typing]
mypy==1.4.1
types-six
types-emoji
types-requests
types-setuptools
types-protobuf
types-tabulate
types-PyYAML

[visual]
matplotlib>=3.2.0
SciencePlots>=2.0.0
