click>=7.0
scikit-learn>=1.2
pandas>=1.5
tqdm>=4.0
joblib>=1.0
numpy>=1.23

[all]
lazypredict[boost,deeplearning,encoders,explain,foundation,interpret,mlflow,timeseries,tune,viz]

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

[deeplearning]
torch>=2.0

[dev]
lazypredict[test]
flake8>=6
mypy>=1.0
build
twine

[encoders]
category_encoders>=2.0

[explain]
shap>=0.42

[flaml]
flaml[automl]>=2.0

[foundation]
timesfm[torch]>=2.0

[interpret]
interpret>=0.4

[mlflow]
mlflow>=2.0.0

[spark]
pyspark>=3.3

[test]
pytest>=7
pytest-cov>=4
xgboost>=1.5
lightgbm>=3.0
statsmodels>=0.13
pmdarima>=2.0
optuna>=3.0
matplotlib>=3.5
category_encoders>=2.0

[timeseries]
statsmodels>=0.13
pmdarima>=2.0

[tune]
optuna>=3.0

[viz]
matplotlib>=3.5
