mlflow>=2.0.0
pydantic>=2.0.0
pandas>=1.3.0
numpy<2.0.0,>=1.20.0

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

[deep-learning]
torch>=1.9.0
pytorch-lightning>=2.0.0
transformers>=4.0.0
sentence-transformers>=2.0.0

[dev]
pytest>=7.0.0
pytest-cov>=3.0.0
black>=22.0.0
mypy>=0.990
ruff>=0.0.260

[lightgbm]
lightgbm>=3.0.0

[pytorch]
torch>=1.9.0
pytorch-lightning>=2.0.0
transformers>=4.0.0
sentence-transformers>=2.0.0

[sklearn]
scikit-learn>=1.0.0

[tabular]
scikit-learn>=1.0.0
xgboost>=1.5.0
lightgbm>=3.0.0

[xgboost]
xgboost>=1.5.0
