numpy>=1.24
pandas>=2.0
sqlalchemy>=2.0

[all]
auto-causal-lib[all-drivers,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]

[bigquery]
sqlalchemy-bigquery>=1.11

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

[clickhouse]
clickhouse-sqlalchemy>=0.3

[cockroachdb]
psycopg2-binary>=2.9

[databricks]
databricks-sqlalchemy>=1.0

[datamine]

[dev]
pytest>=7.0
pyarrow>=14.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]
auto-causal-lib[nlp]
spacy>=3.7

[oracle]
oracledb>=2.0

[parquet]
pyarrow>=14.0

[postgres]
psycopg2-binary>=2.9

[presto]
pyhive[presto]>=0.7

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

[sklearn]
scikit-learn>=1.3

[slm]
torch>=2.0
transformers>=4.36

[snowflake]
snowflake-sqlalchemy>=1.5

[sqlite]

[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

[web]
httpx>=0.27
