fix labels

abcucberkeley · Oct 18, 2023 · f7544c6 · f7544c6
1 parent 6994d79
commit f7544c6
Showing 1 changed file with 5 additions and 1 deletion.
diff --git a/src/eval.py b/src/eval.py
@@ -734,7 +734,11 @@ def plot_heatmap_p2v(
                 transform=ax.transAxes
             )
 
-            ax1.legend(frameon=False, ncol=1, loc='upper left')
+            if color_label == 'Residuals':
+                ax1.legend(frameon=False, ncol=1, loc='upper left')
+            else:
+                ax1.legend(frameon=False, ncol=1, loc='center right')
+
             ax1.yaxis.set_major_formatter(PercentFormatter(decimals=0))
             ax1t.grid(True, which="both", axis='both', lw=.25, ls='--', zorder=0)
             ax2.yaxis.set_major_formatter(PercentFormatter(decimals=0))