pyspark<4.0,>=3.3
pandas>=1.0.0
hyperopt>=0.2.7
plotly>=5.0.0

[all]
smallaxe[xgboost]
smallaxe[lightgbm]
smallaxe[catboost]

[catboost]
catboost>=1.1.0

[dev]
pytest>=7.0.0
pytest-spark>=0.6.0
black>=23.0.0
ruff>=0.1.0
mypy>=1.0.0

[lightgbm]
lightgbm>=3.3.0
synapse-ml-lightgbm>=1.0.0

[xgboost]
xgboost>=1.7.0
