numpy>=1.26.0
PyYAML>=6.0.1
typer>=0.12.5
rich>=13.7.1
sentence-transformers>=3.0.1
transformers<5,>=4.44
einops>=0.8.0

[:python_version < "3.12"]
scikit-learn<1.7

[datafilter]
torch>=2.2.0
accelerate>=0.34.0
safetensors>=0.4.5

[dev]
pytest>=8.0.0
