scikit-learn>=1.0.0
pandas>=1.3.0
numpy>=1.21.0
pyyaml>=6.0
matplotlib>=3.5.0
seaborn>=0.11.0

[all]
nlp-templates[embeddings,mlflow,torch,tuning]

[dev]
pytest>=7.0
pytest-cov>=3.0
black>=22.0
flake8>=4.0
isort>=5.0
mypy>=1.0
pre-commit>=3.0

[docs]
sphinx>=4.0
sphinx-rtd-theme>=1.0
myst-parser>=0.18

[embeddings]
sentence-transformers>=2.2.0

[mlflow]
mlflow>=2.0.0

[torch]
torch>=2.0.0

[tuning]
optuna>=3.0.0
