Divergence heatmap tweak: more s.f. where figure can fit it

Author: AdamGleave

src/evaluating_rewards/analysis/plot_divergence_heatmap.py

@@ -14,6 +14,7 @@
 
 """CLI script to plot heatmap of divergence between pairs of reward models."""
 
+import functools
 import itertools
 import os
 from typing import Any, Iterable, Mapping, Optional
@@ -189,6 +190,7 @@ def hopper():
         for prefix, suffix in itertools.product(activities, MUJOCO_STANDARD_ORDER)
     ]
     heatmap_kwargs["after_plot"] = horizontal_ticks
+    heatmap_kwargs["fmt"] = functools.partial(visualize.short_e, precision=0)
     del activities
 
 

src/evaluating_rewards/analysis/visualize.py

@@ -349,7 +349,7 @@ def compute_mask(
     return res
 
 
-short_fmt = functools.partial(short_e, precision=0)
+short_fmt = functools.partial(short_e, precision=1)
 
 
 def compact_heatmaps(