Fix extraneous spacing between annotation and run time

Author: treysp

sqlmesh/core/console.py

@@ -891,7 +891,7 @@ def update_snapshot_evaluation_progress(
                     self.evaluation_column_widths["duration"]
                 )
 
-                msg = f"{batch} {display_name}  {annotation} {duration}".replace(
+                msg = f"{batch} {display_name}   {annotation}   {duration}".replace(
                     CHECK_MARK, GREEN_CHECK_MARK
                 )
 
@@ -3359,54 +3359,56 @@ def _create_evaluation_model_annotation(snapshot: Snapshot, interval_info: t.Opt
     return interval_info if interval_info else ""
 
 
-def _calculate_interval_str_len(batched_intervals: t.Dict[Snapshot, t.List[Interval]]) -> int:
+def _calculate_interval_str_len(snapshot: Snapshot, intervals: t.List[Interval]) -> int:
     interval_str_len = 0
-    for snapshot, intervals in batched_intervals.items():
-        for interval in intervals:
-            interval_str_len = max(
-                interval_str_len,
-                len(
-                    _create_evaluation_model_annotation(
-                        snapshot, _format_evaluation_model_interval(snapshot, interval)
-                    )
-                ),
-            )
+    for interval in intervals:
+        interval_str_len = max(
+            interval_str_len,
+            len(
+                _create_evaluation_model_annotation(
+                    snapshot, _format_evaluation_model_interval(snapshot, interval)
+                )
+            ),
+        )
     return interval_str_len
 
 
-def _calculate_audit_str_len(batched_intervals: t.Dict[Snapshot, t.List[Interval]]) -> int:
+def _calculate_audit_str_len(snapshot: Snapshot) -> int:
     # The annotation includes audit results. We cannot build the audits result string
     # until after evaluation occurs, but we must determine the annotation column width here.
     # Therefore, we add enough padding for the longest possible audits result string.
     audit_str_len = 0
     audit_base_str_len = len(f", audits ") + 1  # +1 for check/X
-    for snapshot in batched_intervals:
-        if snapshot.is_audit:
+    if snapshot.is_audit:
+        # +1 for "1" audit count, +1 for red X
+        audit_str_len = max(
+            audit_str_len, audit_base_str_len + (2 if not snapshot.audit.blocking else 1)
+        )
+    if snapshot.is_model and snapshot.model.audits:
+        num_audits = len(snapshot.model.audits_with_args)
+        num_nonblocking_audits = sum(
+            1
+            for audit in snapshot.model.audits_with_args
+            if not audit[0].blocking
+            or ("blocking" in audit[1] and audit[1]["blocking"] == exp.false())
+        )
+        if num_audits == 1:
             # +1 for "1" audit count, +1 for red X
-            audit_str_len = max(
-                audit_str_len, audit_base_str_len + (2 if not snapshot.audit.blocking else 1)
-            )
-        if snapshot.is_model and snapshot.model.audits:
-            num_audits = len(snapshot.model.audits_with_args)
-            num_nonblocking_audits = sum(
-                1
-                for audit in snapshot.model.audits_with_args
-                if not audit[0].blocking
-                or ("blocking" in audit[1] and audit[1]["blocking"] == exp.false())
-            )
-            if num_audits == 1:
-                # +1 for "1" audit count, +1 for red X
-                audit_len = audit_base_str_len + (2 if num_nonblocking_audits else 1)
-            else:
-                audit_len = audit_base_str_len + len(str(num_audits))
-                if num_nonblocking_audits:
-                    # +1 for space, +1 for red X
-                    audit_len += len(str(num_nonblocking_audits)) + 2
-            audit_str_len = max(audit_str_len, audit_len)
+            audit_len = audit_base_str_len + (2 if num_nonblocking_audits else 1)
+        else:
+            audit_len = audit_base_str_len + len(str(num_audits))
+            if num_nonblocking_audits:
+                # +1 for space, +1 for red X
+                audit_len += len(str(num_nonblocking_audits)) + 2
+        audit_str_len = max(audit_str_len, audit_len)
     return audit_str_len
 
 
 def _calculate_annotation_str_len(batched_intervals: t.Dict[Snapshot, t.List[Interval]]) -> int:
-    return _calculate_interval_str_len(batched_intervals) + _calculate_audit_str_len(
-        batched_intervals
-    )
+    annotation_str_len = 0
+    for snapshot, intervals in batched_intervals.items():
+        annotation_str_len = max(
+            annotation_str_len,
+            _calculate_interval_str_len(snapshot, intervals) + _calculate_audit_str_len(snapshot),
+        )
+    return annotation_str_len