cosmetics

Author: erindru

sqlmesh/core/console.py

@@ -2270,7 +2270,6 @@ def log_models_updated_during_restatement(
                     dialect=self.dialect,
                 )
                 current_branch = tree.add(display_name)
-                restated_snapshot.table_name()
                 current_branch.add(f"restated version: '{restated_snapshot.version}'")
                 current_branch.add(f"currently active version: '{updated_snapshot.version}'")
 

sqlmesh/core/plan/evaluator.py

@@ -284,11 +284,11 @@ def visit_audit_only_run_stage(
     def visit_restatement_stage(
         self, stage: stages.RestatementStage, plan: EvaluatablePlan
     ) -> None:
-        # Restating intervals on prod plans means that once the data for the intervals being restated has been refreshed
+        # Restating intervals on prod plans means that once the data for the intervals being restated has been backfilled
         # (which happens in the backfill stage) then we need to clear those intervals *from state* across all other environments.
         #
-        # This ensures that work done in dev environments can still be promoted to prod
-        # by forcing dev environments to re-run intervals that changed in prod (because after this stage runs they show as missing)
+        # This ensures that work done in dev environments can still be promoted to prod by forcing dev environments to
+        # re-run intervals that changed in prod (because after this stage runs they are cleared from state and thus show as missing)
         #
         # It also means that any new dev environments created while this restatement plan was running also get the
         # correct intervals cleared because we look up matching snapshots as at right now and not as at the time the plan
@@ -313,8 +313,8 @@ def visit_restatement_stage(
             remove_shared_versions=plan.is_prod,
         )
 
-        # While the restatements were being processed, did any of the snapshots
-        # being restated get new versions deployed? If they did, they will not reflect the data that just got restated
+        # While the restatements were being processed, did any of the snapshots being restated get new versions deployed?
+        # If they did, they will not reflect the data that just got restated, so we need to notify the user
         if deployed_env := self.state_sync.get_environment(plan.environment.name):
             promoted_snapshots_by_name = {s.name: s for s in deployed_env.snapshots}
 

sqlmesh/core/plan/stages.py

@@ -3,9 +3,7 @@
 from dataclasses import dataclass
 from sqlmesh.core import constants as c
 from sqlmesh.core.environment import EnvironmentStatements, EnvironmentNamingInfo, Environment
-from sqlmesh.core.plan.common import (
-    should_force_rebuild,
-)
+from sqlmesh.core.plan.common import should_force_rebuild
 from sqlmesh.core.plan.definition import EvaluatablePlan
 from sqlmesh.core.state_sync import StateReader
 from sqlmesh.core.scheduler import merged_missing_intervals, SnapshotToIntervals
@@ -103,9 +101,9 @@ class RestatementStage:
 
     This stage is effectively a "marker" stage to trigger the plan evaluator to perform the "clear intervals" logic after the BackfillStage has completed.
     The "clear intervals" logic is executed just-in-time using the latest state available in order to pick up new snapshots that may have
-    been created while the BackfillState was running, which is why we do not build a list of snapshots to clear at plan time.
+    been created while the BackfillStage was running, which is why we do not build a list of snapshots to clear at plan time and defer to evaluation time.
 
-    Note that this stage is only present on `prod` plans, because dev plans do not need to worry about clearing intervals in other environments.
+    Note that this stage is only present on `prod` plans because dev plans do not need to worry about clearing intervals in other environments.
 
     Args:
         all_snapshots: All snapshots in the plan by name. Note that this does not include the snapshots from other environments that will get their