Fix: Columns should be sourced from the target table and not the temporary merge table (#4191)

Author: izeigerman

sqlmesh/core/engine_adapter/bigquery.py

@@ -608,7 +608,7 @@ def insert_overwrite_by_partition(
             if columns_to_types is None or columns_to_types[
                 partition_column.name
             ] == exp.DataType.build("unknown"):
-                columns_to_types = self.columns(temp_table_name)
+                columns_to_types = self.columns(table_name)
 
             partition_type_sql = columns_to_types[partition_column.name].sql(dialect=self.dialect)
 

tests/core/engine_adapter/test_bigquery.py

@@ -116,9 +116,7 @@ def test_insert_overwrite_by_partition_query_unknown_column_types(
         },
     )
 
-    columns_mock.assert_called_once_with(
-        exp.to_table(f"test_schema.__temp_test_table_{temp_table_id}")
-    )
+    columns_mock.assert_called_once_with(table_name)
 
     sql_calls = _to_sql_calls(execute_mock)
     assert sql_calls == [