Limit databricks-sql-connector

astronomer · Aug 11, 2023 · e56f3f9 · e56f3f9
1 parent 50aef0b
commit e56f3f9
Showing 1 changed file with 6 additions and 1 deletion.
diff --git a/python-sdk/pyproject.toml b/python-sdk/pyproject.toml
@@ -88,7 +88,10 @@ ftp = [
 openlineage = ["openlineage-airflow>=0.17.0"]
 
 databricks = ["databricks-cli",
-    "apache-airflow-providers-databricks"]
+    "apache-airflow-providers-databricks",
+    # TODO: Remove this once https://github.com/databricks/databricks-sql-python/pull/191 released
+    "databricks-sql-connector<2.9.0",
+]
 
 mssql = [
     "apache-airflow-providers-microsoft-mssql>=3.2",
@@ -115,6 +118,8 @@ all = [
     "sqlalchemy-bigquery>=1.3.0",
     "databricks-cli",
     "apache-airflow-providers-databricks",
+    # TODO: Remove this once https://github.com/databricks/databricks-sql-python/pull/191 released
+    "databricks-sql-connector<2.9.0",
     "s3fs",
     "protobuf",
     "openlineage-airflow>=0.17.0",