tensorflow<3,>=2.19
tensorflow-hub<1,>=0.16
tensorflow-text<3,>=2.19
tf-models-official<3,>=2.19
numpy<2,>=1.26
scikit-learn>=1.3
pandas>=2.0
pyyaml>=6.0

[cohere]
cohere>=5.0
diskcache>=5.0

[dev]
pytest>=9.0
pytest-cov>=7.0
coverage>=7.0
black>=25.0
mypy>=1.19
python-dotenv>=1.0

[huggingface]
transformers<5,>=4.44
huggingface-hub>=0.36
torch<3,>=2.6
diskcache>=5.0

[openai]
openai>=2.0
diskcache>=5.0

[regression]
scipy>=1.10

[text]
tensorflow-text<3,>=2.19

[voyageai]
voyageai>=0.3
diskcache>=5.0
