numpy>=1.23
pandas>=2.0
pyarrow>=14.0
scikit-learn>=1.3

[activelearning-industry]
scikit-activeml>=1.0.0

[ai]
openai>=1.0

[all-classical]
buildml[automl,dashboard,eda,engines,excel,imbalanced,optuna]

[anomaly-industry]
pyod>=1.1
lightgbm>=4.0
xgboost>=2.0

[audio]
buildml[torch]

[automl]
buildml[optuna]

[automl-industry]
buildml[automl]
flaml>=2.1
autogluon.tabular>=1.1
lightgbm>=4.0
xgboost>=2.0
catboost>=1.2

[causal-industry]
dowhy>=0.11
econml>=0.15

[cbr-industry]
hnswlib>=0.8

[dashboard]
fastapi>=0.110
uvicorn[standard]>=0.27
plotly>=5.18
kaleido>=0.2.1
reportlab>=4.0
jinja2>=3.1
python-multipart>=0.0.9

[dev]
pytest>=7.4
pytest-cov>=4.1
ruff>=0.4
mypy>=1.8
build>=1.0
pandas-stubs>=2.0
httpx>=0.27
sphinx
sphinx-rtd-theme
myst-parser
buildml[dashboard]

[dl]
buildml[torch]

[docs]
sphinx
sphinx-rtd-theme
myst-parser

[duckdb]
duckdb>=0.10

[eda]
buildml[reports,viz]

[engines]
buildml[duckdb,polars]

[excel]
openpyxl>=3.1

[federated-industry]
flwr>=1.5

[graph]
networkx>=3.0

[graph-pyg]
buildml[torch]
torch-geometric>=2.5

[imbalanced]
imbalanced-learn>=0.11

[kg-industry]
pykeen>=1.10

[llm]
buildml[ai]

[metalearning-industry]

[metalearning-industry:python_version < "3.13"]
learn2learn>=0.2

[multitask-industry]
lightgbm>=4.0
xgboost>=2.0
catboost>=1.2

[nlp]
nltk>=3.8
langdetect>=1.0.9
sentence-transformers>=2.2
transformers>=4.36

[nlp-industry]
buildml[nlp]
spacy>=3.7

[online-industry]
river>=0.21

[onnx]
onnx>=1.15

[optimize-industry]
pulp>=2.7
ortools>=9.8
cvxpy>=1.4
xgboost>=2.0

[optuna]
optuna>=3.4

[polars]
polars>=0.20

[pretrained]
buildml[speech,vision]

[probabilistic-industry]
mapie>=0.8
ngboost>=0.5

[production]
buildml[torch]
buildml[ssl]
buildml[rag]
buildml[rag-advanced]
buildml[tda]
buildml[unsupervised]
buildml[timeseries]
buildml[timeseries-prophet]
buildml[timeseries-ml]
buildml[graph]
buildml[graph-pyg]
buildml[optuna]
buildml[automl]
buildml[automl-industry]
buildml[tda-industry]
buildml[anomaly-industry]
buildml[semisupervised-industry]
buildml[ranking-industry]
buildml[multitask-industry]
buildml[activelearning-industry]
buildml[metalearning-industry]
buildml[online-industry]
buildml[recommenders-industry]
buildml[recommenders-lightfm]
buildml[causal-industry]
buildml[federated-industry]
buildml[kg-industry]
buildml[probabilistic-industry]
buildml[symbolic-industry]
buildml[cbr-industry]
buildml[nlp]
buildml[nlp-industry]
buildml[optimize-industry]
buildml[synthetic-industry]
buildml[rl]
buildml[rl-industry]

[rag]
sentence-transformers>=2.2
transformers>=4.36

[rag-advanced]
buildml[rag]
langchain-community>=0.2
langchain-core>=0.2

[ranking-industry]
lightgbm>=4.0
xgboost>=2.0
catboost>=1.2

[recommenders-industry]
implicit>=0.7

[recommenders-lightfm]

[recommenders-lightfm:python_version < "3.13" and sys_platform != "win32"]
lightfm>=1.17

[reports]
sweetviz>=2.3
ydata-profiling>=4.6

[rl]
gymnasium>=0.29

[rl-industry]
buildml[rl]
stable-baselines3>=2.3
imitation>=1.0
torch>=2.2

[semisupervised-industry]
lightgbm>=4.0
xgboost>=2.0

[serve]
fastapi>=0.110
uvicorn[standard]>=0.27
python-multipart>=0.0.9
httpx>=0.27
pyyaml>=6.0

[shap]
shap>=0.44

[speech]
buildml[torch]
transformers>=4.36

[ssl]
buildml[torch]
sentence-transformers>=2.2
transformers>=4.36

[symbolic-industry]
imodels>=2.0
z3-solver>=4.12

[symbolic-industry:python_version < "3.13"]
skope-rules>=1.0

[synthetic-industry]
sdv>=1.17
sdmetrics>=0.15

[tda]
ripser>=0.6
persim>=0.3

[tda-industry]
buildml[tda]

[tda-industry:python_version < "3.13"]
giotto-tda>=0.6

[timeseries]
statsmodels>=0.14
scipy>=1.11
ruptures>=1.1

[timeseries-ml]
buildml[timeseries]
torch>=2.2

[timeseries-ml:python_version < "3.13"]
neuralforecast>=1.7

[timeseries-prophet]
buildml[timeseries]
prophet>=1.1

[torch]
torch>=2.2
pillow>=9.0
soundfile>=0.12

[unsupervised]
hdbscan>=0.8
umap-learn>=0.5

[vision]
buildml[torch]
torchvision>=0.17

[viz]
matplotlib>=3.7
seaborn>=0.13
