Pass audit results directly to evaluation prog bar output

Author: treysp

sqlmesh/core/console.py

@@ -30,7 +30,6 @@
 from sqlmesh.core.environment import EnvironmentNamingInfo
 from sqlmesh.core.linter.rule import RuleViolation
 from sqlmesh.core.model import Model
-from sqlmesh.core.model.definition import AuditResult
 from sqlmesh.core.snapshot import (
     Snapshot,
     SnapshotChangeCategory,
@@ -97,12 +96,6 @@ def start_plan_evaluation(self, plan: EvaluatablePlan) -> None:
     def stop_plan_evaluation(self) -> None:
         """Indicates that the evaluation has ended."""
 
-    @abc.abstractmethod
-    def store_evaluation_audit_results(
-        self, snapshot: Snapshot, audit_results: t.List[AuditResult]
-    ) -> None:
-        """Stores the audit results for the snapshot evaluation."""
-
     @abc.abstractmethod
     def start_evaluation_progress(
         self,
@@ -420,11 +413,6 @@ def start_plan_evaluation(self, plan: EvaluatablePlan) -> None:
     def stop_plan_evaluation(self) -> None:
         pass
 
-    def store_evaluation_audit_results(
-        self, snapshot: Snapshot, audit_results: t.List[AuditResult]
-    ) -> None:
-        pass
-
     def start_evaluation_progress(
         self,
         batch_sizes: t.Dict[Snapshot, int],
@@ -739,11 +727,6 @@ def start_plan_evaluation(self, plan: EvaluatablePlan) -> None:
     def stop_plan_evaluation(self) -> None:
         pass
 
-    def store_evaluation_audit_results(
-        self, snapshot: Snapshot, audit_results: t.List[AuditResult]
-    ) -> None:
-        self.evaluation_audit_results[snapshot] = audit_results
-
     def start_evaluation_progress(
         self,
         batch_sizes: t.Dict[Snapshot, int],

sqlmesh/core/scheduler.py

@@ -7,6 +7,7 @@
 from sqlmesh.core.console import Console, get_console
 from sqlmesh.core.environment import EnvironmentNamingInfo, execute_environment_statements
 from sqlmesh.core.macros import RuntimeStage
+from sqlmesh.core.model.definition import AuditResult
 from sqlmesh.core.node import IntervalUnit
 from sqlmesh.core.notification_target import (
     NotificationEvent,
@@ -210,7 +211,6 @@ def evaluate(
             wap_id=wap_id,
             **kwargs,
         )
-        self.console.store_evaluation_audit_results(snapshot, audit_results)
 
         audit_errors_to_raise: t.List[AuditError] = []
         audit_errors_to_warn: t.List[AuditError] = []