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

[:python_version < "3.9"]
typing-extensions

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

[audio]
pystoi>=0.3.0
torchaudio>=0.10.0

[detection]
pycocotools>2.0.0
torchvision>=0.8

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

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

[multimodal]
transformers>=4.10.0

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

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

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

[visual]
matplotlib>=3.2.0
SciencePlots>=2.0.0
