fix: make sqlalchemy optional

Sinaptik-AI · Sep 22, 2023 · c5d8414 · c5d8414
1 parent 0bb9baa
commit c5d8414
Show file tree

Hide file tree

Showing 2 changed files with 44 additions and 72 deletions.
diff --git a/poetry.lock b/poetry.lock
diff --git a/pyproject.toml b/pyproject.toml
@@ -37,6 +37,8 @@ openpyxl = { version = "^3.0.7", optional = true }
 pymysql = { version = "^1.1.0", optional = true }
 psycopg2 = { version = "^2.9.7", optional = true }
 yfinance = { version = "^0.2.28", optional = true }
+sqlalchemy-databricks = { version = "^0.2.0", optional = true }
+snowflake-sqlalchemy = { version = "^1.5.0", optional = true }
 
 [tool.poetry.group.dev.dependencies]
 black = "^23.3.0"
@@ -50,12 +52,10 @@ coverage = "^7.2.7"
 
 [tool.poetry.group.extras.dependencies]
 google-cloud-aiplatform = "^1.26.1"
-sqlalchemy-databricks = "^0.2.0"
-snowflake-sqlalchemy = "^1.5.0"
 
 [tool.poetry.extras]
 connectors = [ "pymysql", "psycopg2"]
-google-ai = ["google-generativeai", "google-cloud-aiplatform"]
+google-ai = ["google-generativeai", "google-cloud-aiplatform", "sqlalchemy-databricks", "snowflake-sqlalchemy"]
 google-sheets = ["beautifulsoup4"]
 excel = ["openpyxl"]
 polars = ["polars"]