numpy>1.20.0
torch>=1.8.1
lightning-utilities>=0.8.0

[:python_version < "3.9"]
typing-extensions

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

[audio]
pystoi>=0.3.0
torchaudio>=0.10.0

[detection]
torchvision>=0.8
pycocotools>2.0.0

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

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

[multimodal]
piq<=0.8.0
transformers>=4.10.0

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

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

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

[visual]
SciencePlots>=2.0.0
matplotlib>=3.2.0
