tweaks

Author: erindru

sqlmesh/core/plan/explainer.py

@@ -90,18 +90,16 @@ def from_restatement_stage(
         state_reader: StateReader,
         plan: EvaluatablePlan,
     ) -> ExplainableRestatementStage:
-        loaded_snapshots = {s.snapshot_id: s for s in stage.all_snapshots.values()}
-
         all_restatement_intervals = identify_restatement_intervals_across_snapshot_versions(
             state_reader=state_reader,
             prod_restatements=plan.restatements,
             disable_restatement_models=plan.disabled_restatement_models,
-            loaded_snapshots=loaded_snapshots,
+            loaded_snapshots={s.snapshot_id: s for s in stage.all_snapshots.values()},
         )
 
         # Group the interval clear requests by snapshot name to make them easier to write to the console
         snapshot_intervals_to_clear = defaultdict(list)
-        for _, clear_request in all_restatement_intervals.items():
+        for clear_request in all_restatement_intervals.values():
             snapshot_intervals_to_clear[clear_request.snapshot.name].append(clear_request)
 
         return cls(
@@ -216,20 +214,20 @@ def visit_restatement_stage(
         if isinstance(stage, ExplainableRestatementStage) and (
             snapshot_intervals := stage.snapshot_intervals_to_clear
         ):
-            for name, requests in snapshot_intervals.items():
+            for name, clear_requests in snapshot_intervals.items():
                 display_name = model_display_name(
                     name, self.environment_naming_info, self.default_catalog, self.dialect
                 )
-                interval_start = min(cr.interval[0] for cr in requests)
-                interval_end = max(cr.interval[1] for cr in requests)
+                interval_start = min(cr.interval[0] for cr in clear_requests)
+                interval_end = max(cr.interval[1] for cr in clear_requests)
 
                 if not interval_start or not interval_end:
                     continue
 
                 node = tree.add(f"{display_name} [{to_ts(interval_start)} - {to_ts(interval_end)}]")
 
                 all_environment_names = sorted(
-                    set(env_name for cr in requests for env_name in cr.environment_names)
+                    set(env_name for cr in clear_requests for env_name in cr.environment_names)
                 )
                 node.add("in environments: " + ", ".join(all_environment_names))