PyYAML>=6.0
hydra-core>=1.3
omegaconf>=2.3
lightning>=2.4
torch>=2.3
numpy>=1.26
matplotlib>=3.8
pillow>=10.0
pandas>=2.0
scikit-learn
onnx>=1.16
onnxscript
onnxruntime>=1.18
rich>=15.0

[all]
pyteia[audio,fid,tabular,vision]

[audio]
torchaudio
soundfile

[fid]
pytorch-fid==0.3.0

[tabular]
sqlalchemy
pyarrow

[vision]
torchvision
timm
