wip

allenai · Feb 20, 2024 · 21dc9cd · 21dc9cd
1 parent cc9298d
commit 21dc9cd
Showing 1 changed file with 2 additions and 1 deletion.
diff --git a/herm/visualization.py b/herm/visualization.py
@@ -85,7 +85,8 @@ def draw_per_token_reward(
     title = "Cumulative substring rewards"
     ax.set_title(title, pad=20)
 
-    fig.tight_layout()
+    # fig.tight_layout()
+    fig.subplots_adjust(left=0.5)
     if output_path:
         print(f"Saving per-token-reward heatmap to {output_path}")
         plt.savefig(output_path, transparent=True, dpi=120)