From 51aad90f0ab554fb5b39297128c1f2bb1c9c1f25 Mon Sep 17 00:00:00 2001 From: Vincent Chan Date: Fri, 19 Dec 2025 11:19:51 -0800 Subject: [PATCH] Chore: Update type annotation for pandas and pandas-stubs >= v2.3.3 --- sqlmesh/core/test/definition.py | 5 +++-- 1 file changed, 3 insertions(+), 2 deletions(-) diff --git a/sqlmesh/core/test/definition.py b/sqlmesh/core/test/definition.py index 1c9807cfa1..8694ec6024 100644 --- a/sqlmesh/core/test/definition.py +++ b/sqlmesh/core/test/definition.py @@ -355,11 +355,12 @@ def _to_hashable(x: t.Any) -> t.Any: for df in _split_df_by_column_pairs(diff) ) else: - from pandas import MultiIndex + from pandas import DataFrame, MultiIndex levels = t.cast(MultiIndex, diff.columns).levels[0] for col in levels: - col_diff = diff[col] + # diff[col] returns a DataFrame when columns is a MultiIndex + col_diff = t.cast(DataFrame, diff[col]) if not col_diff.empty: table = df_to_table( f"[bold red]Column '{col}' mismatch{failed_subtest}[/bold red]",