click
scikit-learn>=1.0
pandas>=1.3
tqdm>=4.0
joblib>=1.0
numpy>=1.21

[all]
xgboost>=1.5
lightgbm>=3.0
catboost>=1.0
mlflow<3.0,>=2.0.0
category_encoders>=2.0
statsmodels>=0.13
pmdarima>=2.0
torch>=2.0
timesfm
optuna>=3.0
shap>=0.42
interpret>=0.4
matplotlib>=3.5

[boost]
xgboost>=1.5
lightgbm>=3.0
catboost>=1.0

[deeplearning]
torch>=2.0

[dev]
pytest>=7
pytest-cov>=4
flake8>=6
mypy>=1.0
optuna>=3.0

[explain]
shap>=0.42

[flaml]
flaml>=2.0

[foundation]
timesfm

[interpret]
interpret>=0.4

[mlflow]
mlflow<3.0,>=2.0.0

[spark]
pyspark>=3.3

[timeseries]
statsmodels>=0.13
pmdarima>=2.0

[tune]
optuna>=3.0

[viz]
matplotlib>=3.5
