
[all]
spacy<4.0,>=3.7
transformers>=4.40
sentencepiece
accelerate
rank_bm25
sentence-transformers>=2.7
wandb>=0.16

[all:platform_system != "Darwin" or platform_machine != "arm64"]
torch>=2.0

[dev]
pytest
ruff
mypy
pre-commit

[nli]
transformers>=4.40
sentencepiece
accelerate
rank_bm25

[nli:platform_system != "Darwin" or platform_machine != "arm64"]
torch>=2.0

[rerank]
sentence-transformers>=2.7

[unitize]
spacy<4.0,>=3.7

[wandb]
wandb>=0.16
