Improve message formatting

Author: treysp

sqlmesh/core/console.py

@@ -3817,7 +3817,7 @@ def update_snapshot_evaluation_progress(
         message = f"Evaluated {snapshot.name} | batch={batch_idx} | duration={duration_ms}ms | num_audits_passed={num_audits_passed} | num_audits_failed={num_audits_failed}"
 
         if auto_restatement_triggers:
-            message += f" | Auto-restatement triggers {', '.join(trigger.name for trigger in auto_restatement_triggers)}"
+            message += f" | auto_restatement_triggers=[{', '.join(trigger.name for trigger in auto_restatement_triggers)}]"
 
         if audit_only:
             message = f"Audited {snapshot.name} | duration={duration_ms}ms | num_audits_passed={num_audits_passed} | num_audits_failed={num_audits_failed}"

sqlmesh/core/scheduler.py

@@ -29,7 +29,6 @@
 from sqlmesh.core.snapshot.definition import check_ready_intervals
 from sqlmesh.core.snapshot.definition import (
     Interval,
-    SnapshotIntervals,
     expand_range,
     parent_snapshots_by_name,
 )
@@ -642,17 +641,15 @@ def _run_or_audit(
         for s_id, interval in (remove_intervals or {}).items():
             self.snapshots[s_id].remove_interval(interval)
 
-        auto_restated_intervals: t.List[SnapshotIntervals] = []
-        auto_restatement_triggers: t.Dict[SnapshotId, t.List[SnapshotId]] = {}
+        all_auto_restatement_triggers: t.Dict[SnapshotId, t.List[SnapshotId]] = {}
         if auto_restatement_enabled:
-            auto_restated_intervals, auto_restatement_triggers = apply_auto_restatements(
+            auto_restated_intervals, all_auto_restatement_triggers = apply_auto_restatements(
                 self.snapshots, execution_time
             )
             self.state_sync.add_snapshots_intervals(auto_restated_intervals)
             self.state_sync.update_auto_restatements(
                 {s.name_version: s.next_auto_restatement_ts for s in self.snapshots.values()}
             )
-        auto_restated_snapshots = {snapshot.snapshot_id for snapshot in auto_restated_intervals}
 
         merged_intervals = self.merged_missing_intervals(
             start,
@@ -670,9 +667,8 @@ def _run_or_audit(
             return CompletionStatus.NOTHING_TO_DO
 
         merged_intervals_snapshots = {snapshot.snapshot_id for snapshot in merged_intervals}
-
-        auto_restatement_triggers_dict: t.Dict[SnapshotId, t.List[SnapshotId]] = {
-            s_id: auto_restatement_triggers.get(s_id, []) for s_id in merged_intervals_snapshots
+        auto_restatement_triggers: t.Dict[SnapshotId, t.List[SnapshotId]] = {
+            s_id: all_auto_restatement_triggers.get(s_id, []) for s_id in merged_intervals_snapshots
         }
 
         errors, _ = self.run_merged_intervals(
@@ -686,7 +682,7 @@ def _run_or_audit(
             run_environment_statements=run_environment_statements,
             audit_only=audit_only,
             restatements=remove_intervals,
-            auto_restatement_triggers=auto_restatement_triggers_dict,
+            auto_restatement_triggers=auto_restatement_triggers,
         )
 
         return CompletionStatus.FAILURE if errors else CompletionStatus.SUCCESS