Preserve slice titles and colorbar labels in saved figure bounds

2026-09-19 17:05:46 -04:00
parent 07799bca34
commit 2c6523b1ce
2 changed files with 4 additions and 2 deletions
--- a/experiments/compare_model_slices.py
+++ b/experiments/compare_model_slices.py
@@ -152,7 +152,7 @@ def main() -> None:
    args.output.parent.mkdir(parents=True, exist_ok=True)
    paths[0].write_text(json.dumps(report, indent=2, allow_nan=False) + "\n")
    for path in paths[1:]:
-        figure.savefig(path, dpi=200)
+        figure.savefig(path, dpi=200, bbox_inches="tight", pad_inches=.15)
    plt.close(figure)
    for path in paths:
        print(path)