speed up execution by moving row_diff in concurrent method

Author: themisvaltinos

sqlmesh/core/console.py

@@ -2139,7 +2139,6 @@ def show_row_diff(
                 self.console.print(f"\n[b][green]{target_name} ONLY[/green] sample rows:[/b]")
                 self.console.print(row_diff.t_sample.to_string(index=False), end="\n\n")
 
-
     def show_table_diff(
         self,
         table_diffs: t.List[TableDiff],

sqlmesh/core/context.py

@@ -1644,6 +1644,9 @@ def table_diff(
                     on=on,
                     skip_columns=skip_columns,
                     where=where,
+                    show=show,
+                    temp_schema=temp_schema,
+                    skip_grain_check=skip_grain_check,
                 ),
                 tasks_num=tasks_num,
             )
@@ -1678,6 +1681,9 @@ def _model_diff(
         on: t.Optional[t.List[str] | exp.Condition] = None,
         skip_columns: t.Optional[t.List[str]] = None,
         where: t.Optional[str | exp.Condition] = None,
+        show: bool = True,
+        temp_schema: t.Optional[str] = None,
+        skip_grain_check: bool = False,
     ) -> TableDiff:
         model = self.get_model(model_or_snapshot, raise_if_missing=True)
         adapter = self._get_engine_adapter(model.gateway)
@@ -1704,7 +1710,7 @@ def _model_diff(
                     # Handle a single Column or Paren expression
                     on.append(expr.this.sql(dialect=adapter.dialect))
 
-        return self._table_diff(
+        table_diff = self._table_diff(
             on=on,
             skip_columns=skip_columns,
             where=where,
@@ -1717,6 +1723,11 @@ def _model_diff(
             source_alias=source_alias,
             target_alias=target_alias,
         )
+        # Trigger row_diff in parallel execution so it's available for ordered display later
+        if show:
+            table_diff.row_diff(temp_schema=temp_schema, skip_grain_check=skip_grain_check)
+
+        return table_diff
 
     def _table_diff(
         self,