scikit-learn>=1.0.0
pandas>=1.0.0
diskcache>=5.6.1
skops>=0.8.0
model2vec

[all]
sentence-transformers>=2.2.2
timm>=0.6.7
openai>=1.59.8

[cohere]
cohere>=4.11.2

[dev]
embetter[all,docs]
interrogate>=1.5.0
pytest>=4.0.2
ruff
pre-commit>=2.2.0
mktestdocs==0.2.4
datasets==2.8.0
pyarrow==20.0.0
matplotlib
pytest-xdist

[docs]
mkdocs-material==9.6.9
mkdocstrings==0.29.0
mkdocstrings-python==1.16.0
mktestdocs==0.2.4

[ollama]
ollama>=0.5.3

[openai]
openai>=1.59.8

[pytorch]
torch>=1.12.0

[sbert]
sentence-transformers>=2.2.2

[text]
sentence-transformers>=2.2.2

[vision]
timm>=0.6.7
