scikit-learn>=0.22.2
pandas>=1.0.0
matplotlib>=3.1.1
scipy>=1.4.0
joblib>=0.13.2
tqdm>=4.41.0
shap<0.43.0,>=0.41.0
numpy>=1.23.2
numba>=0.57.0

[all]
probatus[dev,docs,extras]

[dev]
black>=19.10b0
pre-commit>=2.5.0
mypy>=0.770
pytest>=6.0.0
pytest-cov>=2.10.0
pyflakes
seaborn>=0.9.0
joblib>=0.13.2
jupyter>=1.0.0
tabulate>=0.8.7
nbconvert>=6.0.7
pre-commit>=2.7.1
isort>=5.12.0
codespell>=2.2.4
ruff>=0.2.2

[docs]
mkdocs>=1.5.3
mkdocs-jupyter>=0.24.3
mkdocs-material>=9.5.13
mkdocstrings>=0.24.1
mkdocstrings-python>=1.8.0

[extras]
lightgbm>=3.3.0
xgboost>=1.5.0
scipy>=1.4.0

[extras:python_version != "3.8"]
catboost>=1.1

[extras:python_version == "3.8"]
catboost<1.2
