#------- Core dependencies -------
torch>=2.3,<2.4

transformers>=4.41,<4.42
datasets>=2.19,<2.20
rich>=13.0.0,<14.0.0
scikit-learn>=1.3,<1.5
typer>=0.12,<0.13
mosaicml>=0.22,<0.23

# train
lightning>=2.2,<2.4
mosaicml-streaming>=0.7.5,<0.8
hydra-core>=1.3,<1.4
hydra_colorlog
wandb>=0.16,<0.17

# utils
art==6.1
pprintpp==0.4.0
colorama==0.4.6

#------- Optional dependencies -------

# faiss
faiss-cpu==1.8.0 # needed by: faiss

# onnx GPU
optimum[onnxruntime-gpu]>=1.7,<1.8; platform_system == "Linux" # needed by: onnx-gpu
termcolor>=2.3,<2.5 # needed by: onnx-gpu

# onnx CPU
optimum[onnxruntime]>=1.14,<1.17 # needed by: onnx
termcolor>=2.3,<2.5 # needed by: onnx

# serve
fastapi>=0.104,<0.110 # needed by: serve
uvicorn[standard]>=0.24,<0.28 # needed by: serve
gunicorn==21.2.0 # needed by: serve
ray[serve]>=2.8,<=2.10 # needed by: serve

# dev
pre-commit # needed by: dev
black[d] # needed by: dev
isort # needed by: dev
