diff --git a/generated/provider_dependencies.json b/generated/provider_dependencies.json index dc7aedc3b980b..75005dacb819e 100644 --- a/generated/provider_dependencies.json +++ b/generated/provider_dependencies.json @@ -434,7 +434,7 @@ "aiohttp>=3.9.2, <4", "apache-airflow-providers-common-sql>=1.10.0", "apache-airflow>=2.8.0", - "databricks-sql-connector>=2.0.0, <3.0.0, !=2.9.0", + "databricks-sql-connector>=3.0.0", "mergedeep>=1.3.4", "pandas>=1.5.3,<2.2;python_version<\"3.9\"", "pandas>=2.1.2,<2.2;python_version>=\"3.9\"", diff --git a/providers/src/airflow/providers/databricks/provider.yaml b/providers/src/airflow/providers/databricks/provider.yaml index fe2bb9f7fde45..ff35323617845 100644 --- a/providers/src/airflow/providers/databricks/provider.yaml +++ b/providers/src/airflow/providers/databricks/provider.yaml @@ -74,10 +74,7 @@ dependencies: - apache-airflow>=2.8.0 - apache-airflow-providers-common-sql>=1.10.0 - requests>=2.27.0,<3 - # The connector 2.9.0 released on Aug 10, 2023 has a bug that it does not properly declare urllib3 and - # it needs to be excluded. See https://github.com/databricks/databricks-sql-python/issues/190 - # The 2.9.1 (to be released soon) already contains the fix - - databricks-sql-connector>=2.0.0, <3.0.0, !=2.9.0 + - databricks-sql-connector>=3.0.0 - aiohttp>=3.9.2, <4 - mergedeep>=1.3.4 - pandas>=2.1.2,<2.2;python_version>="3.9" diff --git a/providers/tests/snowflake/operators/test_snowflake_sql.py b/providers/tests/snowflake/operators/test_snowflake_sql.py index fb1bcd1726352..47dcf27122df1 100644 --- a/providers/tests/snowflake/operators/test_snowflake_sql.py +++ b/providers/tests/snowflake/operators/test_snowflake_sql.py @@ -32,9 +32,11 @@ except ImportError: # Row is used in the parametrize so it's parsed during collection and we need to have a viable # replacement for the collection time when databricks is not installed (Python 3.12 for now) - def Row(*args, **kwargs): + def MockRow(*args, **kwargs): return MagicMock() + Row = MockRow + from airflow.models.connection import Connection from airflow.providers.common.compat.openlineage.facet import (