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

[:python_version < "3.9"]
typing-extensions

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

[audio]
torchaudio>=0.10.0
pystoi>=0.3.0

[detection]
torchvision>=0.8
pycocotools>2.0.0

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

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

[multimodal]
transformers>=4.10.0

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

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

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

[visual]
matplotlib>=3.2.0
SciencePlots>=2.0.0
