click>=8.1.7
scipy>=1.10.1
wandb>=0.16.3
protobuf>=3.20.2
urllib3<2,>=1.26.18
pydantic>=2.6.4
pydantic-yaml>=1.2.0
pydantic==2.6.4
pydantic-yaml==1.2.0
ray[default]==2.30.0
loguru==0.7.2
s3fs

[dev]
lm-buddy[docs,jobs,ruff,test]
pre-commit==3.6.0
jupyter==1.0.0

[docs]
Sphinx==7.2.6
nbsphinx==0.9.3
sphinx-codeautolink==0.15.0
myst-parser==2.0.0
furo==2024.1.29

[jobs]
torch==2.3.1
datasets>=2.17.1
transformers==4.38.0
accelerate==0.26.1
peft==0.7.1
trl==0.7.10
bitsandbytes==0.42.0
bert_score==0.3.13
lm-eval==0.4.2
einops==0.7.0
openai==1.14.3
ragas==0.1.5
langchain-community==0.0.29
langchain_openai==0.1.1
sentencepiece==0.2.0
evaluate==0.4.2

[ruff]
ruff==0.2.1

[test]
pytest==7.4.3
pytest-cov==4.1.0
