diff --git a/sqlmesh/core/engine_adapter/databricks.py b/sqlmesh/core/engine_adapter/databricks.py index bb671822e3..dbf38f0b94 100644 --- a/sqlmesh/core/engine_adapter/databricks.py +++ b/sqlmesh/core/engine_adapter/databricks.py @@ -431,6 +431,7 @@ def columns( .order_by("ordinal_position ASC") ) - result = self.cursor.fetchall(query) + self.cursor.execute(query) + result = self.cursor.fetchall() return {row[0]: exp.DataType.build(row[1], dialect=self.dialect) for row in result} diff --git a/tests/core/engine_adapter/test_databricks.py b/tests/core/engine_adapter/test_databricks.py index 9312616a54..42cbd287f2 100644 --- a/tests/core/engine_adapter/test_databricks.py +++ b/tests/core/engine_adapter/test_databricks.py @@ -578,7 +578,7 @@ def test_columns(mocker: MockFixture, make_mocked_engine_adapter: t.Callable): ), } - adapter.cursor.fetchall.assert_called_once_with( + adapter.cursor.execute.assert_called_once_with( parse_one( """SELECT columns.column_name, columns.full_data_type FROM system.information_schema.columns WHERE table_name = 'test_table' AND table_schema = 'test_db' AND table_catalog = 'test_catalog' ORDER BY ordinal_position ASC""", dialect="databricks",