mlflow>=2.0.0
pydantic>=2.0.0
pandas>=1.3.0
numpy<2.0.0,>=1.20.0
scikit-learn>=1.0.0
xgboost>=1.5.0
torch>=1.9.0
pytorch-lightning>=2.0.0
lightgbm>=3.0.0
transformers>=4.0.0
sentence-transformers>=2.0.0

[all]
scikit-learn>=1.0.0
xgboost>=1.5.0
torch>=1.9.0
pytorch-lightning>=2.0.0
lightgbm>=3.0.0
transformers>=4.0.0
sentence-transformers>=2.0.0
PyYAML>=5.4.0
toml>=0.10.0

[config]
PyYAML>=5.4.0
toml>=0.10.0

[dev]
pytest>=6.0.0
pytest-cov>=3.0.0
pytest-mock>=3.6.0
black>=22.0.0
isort>=5.10.0
flake8>=4.0.0
mypy>=0.950
pre-commit>=2.15.0
twine>=4.0.0
build>=0.7.0

[docs]
sphinx>=4.0.0
sphinx-rtd-theme>=1.0.0
sphinx-autodoc-typehints>=1.12.0
myst-parser>=0.18.0

[lightgbm]
lightgbm>=3.0.0

[pytorch]
torch>=1.9.0
pytorch-lightning>=2.0.0

[sklearn]
scikit-learn>=1.0.0

[test]
pytest>=6.0.0
pytest-cov>=3.0.0
pytest-mock>=3.6.0

[transformers]
transformers>=4.0.0
sentence-transformers>=2.0.0

[xgboost]
xgboost>=1.5.0
