numpy<3.0.0,>=1.24.0
scikit-learn<2.0.0,>=1.3.0
torch<3.0.0,>=2.6.0
pandas<3.0.0,>=2.0.0
tqdm<5.0.0,>=4.66.0

[dev]
pytest>=8.0.0
pytest-cov>=5.0.0
black>=24.0.0
isort>=5.13.0
flake8>=7.0.0
mypy>=1.9.0
vulture>=2.11
pydocstyle>=6.3.0
codespell>=2.3.0
pre-commit>=3.7.0
bandit>=1.7.8
pip-audit>=2.7.0
wheel>=0.43.0
build>=1.2.0

[docs]
sphinx>=7.3.0
sphinx-rtd-theme>=2.0.0
myst-parser>=3.0.0

[transformers]
transformers<5.0.0,>=4.53.0
datasets<4.0.0,>=2.18.0
