numpy>=1.24
pandas>=2.0
sqlalchemy>=2.0
scikit-learn>=1.3
nltk>=3.8
httpx>=0.27
reportlab>=4.0
pyarrow>=14.0
torch>=2.0
transformers>=4.36
accelerate>=0.27
huggingface_hub>=0.20
langgraph>=0.2
langchain-core>=0.2
mcp>=1.0

[all]
auto-causal-lib[all-drivers,bitsandbytes,causal-extra,mcp,ml,nlp,parquet,slm,ui,web]

[all-drivers]
auto-causal-lib[bigquery,clickhouse,cockroachdb,databricks,duckdb,mariadb,mssql,mysql,oracle,parquet,postgres,presto,redshift,snowflake,synapse,trino,vertica]

[all-safe]
scipy
statsmodels
patsy
linearmodels
arch
lifelines
scikit-posthocs
scikit-learn>=1.3
optuna
imbalanced-learn
shap
spacy>=3.7
sentence-transformers
textblob
keybert
causal-learn>=0.1.3.8
dowhy>=0.11
DoubleML>=0.7
lingam>=1.8
pgmpy
pandera
polars
pyarrow>=14.0
matplotlib
seaborn
plotly>=5.18
kaleido
networkx

[automl]
scikit-learn>=1.3
xgboost
lightgbm
catboost
optuna
imbalanced-learn
shap

[bigquery]
sqlalchemy-bigquery>=1.11

[bitsandbytes]
bitsandbytes>=0.43

[causal-extra]
dowhy>=0.11
econml>=0.15
causal-learn>=0.1.3.8
DoubleML>=0.7
lingam>=1.8
gcastle>=1.0.3
pgmpy
PySensemakr
linearmodels
scikit-learn>=1.3

[clickhouse]
clickhouse-sqlalchemy>=0.3

[cockroachdb]
psycopg2-binary>=2.9

[databricks]
databricks-sqlalchemy>=1.0

[datamine]

[dev]
pytest>=7.0
twine>=5.0
build>=1.0

[duckdb]
duckdb>=0.10
duckdb-engine>=0.13

[firebird]
sqlalchemy-firebird>=0.8

[hana]
sqlalchemy-hana>=0.5

[isolates]

[mariadb]
pymysql>=1.1

[mcp]
mcp>=1.0

[ml]
auto-causal-lib[sklearn,torch]

[mssql]
pyodbc>=5.0

[mysql]
pymysql>=1.1

[nlp]
nltk>=3.8
gensim>=4.3

[nlp-full]
nltk>=3.8
spacy>=3.7
sentence-transformers
stanza
textblob
keybert
chromadb

[nlp-full:platform_system != "Windows"]
faiss-cpu

[oracle]
oracledb>=2.0

[parquet]
pyarrow>=14.0

[postgres]
psycopg2-binary>=2.9

[presto]
pyhive[presto]>=0.7

[production]
auto-causal-lib[stats,viz]
pandera
polars
pyarrow>=14.0
imbalanced-learn
mlflow

[redshift]
sqlalchemy-redshift>=0.8
psycopg2-binary>=2.9

[research]
auto-causal-lib[automl,causal-extra,nlp-full,stats,viz]
pymc
arviz
great-expectations
ydata-profiling

[sklearn]
scikit-learn>=1.3

[slm]
torch>=2.0
transformers>=4.36
accelerate>=0.27
huggingface_hub>=0.20

[snowflake]
snowflake-sqlalchemy>=1.5

[sqlite]

[stats]
scipy
statsmodels
patsy
linearmodels
arch
lifelines
scikit-posthocs

[streamlit]
auto-causal-lib[ui]

[suite]
auto-causal-lib[ml,nlp,slm,ui,web]

[synapse]
pyodbc>=5.0

[torch]
torch>=2.0

[trino]
trino>=0.328

[ui]
streamlit>=1.28
plotly>=5.18

[vertica]
vertica-python>=1.3

[viz]
matplotlib
seaborn
plotly>=5.18
kaleido
networkx
graphviz

[web]
httpx>=0.27
