xgboost<4.0,>=2.0
torch<2.10,>=2.1
pyro-ppl<1.10,>=1.8
numpy<2.4,>=1.26.0
pandas<2.3,>=2.0
scipy<1.17,>=1.11
tqdm<4.68,>=4.66
matplotlib<3.11,>=3.8
scikit-base

[all_extras]
optuna<4.4,>=3.5.0
optuna-integration
seaborn<0.14.0,>=0.13.0

[all_extras:python_version < "3.14"]
shap>=0.50.0

[docs]
mkdocs
mkdocstrings[python]
mkdocs-jupyter

[notebooks]
scikit-learn<1.8.0,>=1.4.0
plotnine<0.16.0,>=0.12.0

[test]
flake8
pytest
pytest-coverage
pytest-timeout
