Remove weighted from main heatmap figures

scripts/make_figures.py

@@ -339,7 +339,7 @@ def fig1_allocation_geometry(suite: dict[str, dict], output_dir: Path):
 
 def fig1_disparity_heatmap(suite: dict[str, dict], output_dir: Path):
     """Heatmap of max disparity across regimes and methods."""
-    methods = METHOD_ORDER
+    methods = [method for method in METHOD_ORDER if method != "weighted"]
     matrix = np.full((len(methods), len(DGP_SPECS)), np.nan)
 
     for j, (stem, _) in enumerate(DGP_SPECS):
@@ -383,7 +383,7 @@ def fig1_disparity_heatmap(suite: dict[str, dict], output_dir: Path):
                 path_effects=[pe.withStroke(linewidth=1.1, foreground=stroke_color)],
             )
 
-    highlight_best_cells(ax, matrix, methods, exclude={"oracle", "weighted"})
+    highlight_best_cells(ax, matrix, methods, exclude={"oracle"})
 
     cbar = fig.colorbar(im, ax=ax, shrink=0.95, pad=0.02)
     cbar.set_label("Max disparity")
@@ -529,7 +529,7 @@ def fig4_runtime_tradeoff(suite: dict[str, dict], output_dir: Path):
 
 
 def fig5_real_disparity_heatmap(real_suite: dict[str, dict], output_dir: Path):
-    methods = ["global", "partition", "twostage", "fullcp", "jackknife_plus", "oneshot", "trainres", "weighted"]
+    methods = ["global", "partition", "twostage", "fullcp", "jackknife_plus", "oneshot", "trainres"]
     tasks = [task for _, task in REAL_SPECS if task in real_suite]
     if not tasks:
         print("Skipping Fig 5: no real-data results found")
@@ -584,7 +584,7 @@ def fig5_real_disparity_heatmap(real_suite: dict[str, dict], output_dir: Path):
                 path_effects=[pe.withStroke(linewidth=1.1, foreground=stroke_color)],
             )
 
-    highlight_best_cells(ax, matrix, methods, exclude={"weighted"})
+    highlight_best_cells(ax, matrix, methods)
 
     cbar = fig.colorbar(im, ax=ax, shrink=0.95, pad=0.02)
     cbar.set_label("Max disparity")