diff --git a/.github/workflows/test_local_sources.yml b/.github/workflows/test_local_sources.yml index 1707e187e0..818c818ff2 100644 --- a/.github/workflows/test_local_sources.yml +++ b/.github/workflows/test_local_sources.yml @@ -84,7 +84,7 @@ jobs: # TODO: which deps should we enable? - name: Install dependencies - run: poetry install --no-interaction -E postgres -E duckdb -E parquet -E filesystem -E cli -E weaviate -E qdrant --with sentry-sdk --with pipeline -E deltalake + run: poetry install --no-interaction -E postgres -E duckdb -E parquet -E filesystem -E cli -E sql_database --with sentry-sdk --with pipeline # we need sqlalchemy 2 for the sql_database tests - name: Upgrade sql alchemy diff --git a/poetry.lock b/poetry.lock index 745f1c63f9..94859769e0 100644 --- a/poetry.lock +++ b/poetry.lock @@ -9738,10 +9738,11 @@ qdrant = ["qdrant-client"] redshift = ["psycopg2-binary", "psycopg2cffi"] s3 = ["botocore", "s3fs"] snowflake = ["snowflake-connector-python"] +sql-database = ["connectorx", "pymysql", "sqlalchemy"] synapse = ["adlfs", "pyarrow", "pyodbc"] weaviate = ["weaviate-client"] [metadata] lock-version = "2.0" python-versions = ">=3.8.1,<3.13" -content-hash = "ee2aee14ef4cd198e8f6fb35a35305fbe6d02650b4c71e45c625ad83556e2c95" +content-hash = "b4ef2e842b43b2da1b0594fc644c9708d7223f8d4e22c35e6e9e3ad1e1f0bebe" diff --git a/pyproject.toml b/pyproject.toml index 52ae94a2b4..8fdaf987cc 100644 --- a/pyproject.toml +++ b/pyproject.toml @@ -83,6 +83,10 @@ lancedb = { version = ">=0.8.2", optional = true, markers = "python_version >= ' tantivy = { version = ">= 0.22.0", optional = true } deltalake = { version = ">=0.19.0", optional = true } graphlib-backport = {version = "*", python = "<3.9"} +sqlalchemy = { version = ">=1.4", optional = true } +pymysql = { version = "^1.0.3", optional = true } +connectorx = { version = ">=0.3.1", optional = true } + [tool.poetry.extras] gcp = ["grpcio", "google-cloud-bigquery", "db-dtypes", "gcsfs"] @@ -109,6 +113,7 @@ clickhouse = ["clickhouse-driver", "clickhouse-connect", "s3fs", "gcsfs", "adlfs dremio = ["pyarrow"] lancedb = ["lancedb", "pyarrow", "tantivy"] deltalake = ["deltalake", "pyarrow"] +sql_database = ["sqlalchemy", "pymysql", "connectorx"] [tool.poetry.scripts] @@ -159,11 +164,6 @@ types-regex = "^2024.5.15.20240519" flake8-print = "^5.0.0" mimesis = "^7.0.0" -[tool.poetry.group.sql_database.dependencies] -sqlalchemy = ">=1.4" -pymysql = "^1.0.3" -connectorx = ">=0.3.1" - [tool.poetry.group.pipeline] optional = true