From 01411e09ef20e4f420aa4f32b97081bfbd3845a8 Mon Sep 17 00:00:00 2001
From: Theo Heimel <info@theovention.de>
Date: Wed, 3 May 2023 09:43:00 -0400
Subject: [PATCH] update labels

---
 src/plots.py | 16 ++++++++--------
 1 file changed, 8 insertions(+), 8 deletions(-)

diff --git a/src/plots.py b/src/plots.py
index 224cf32..84f9ba3 100644
--- a/src/plots.py
+++ b/src/plots.py
@@ -398,8 +398,8 @@ def plot_bgen_weights(self, file: str):
             )
             #cb = plt.colorbar(img)
             #cb.set_label('norm.')
-            ax.set_xlabel(r"$\sigma(p_\text{gen}) / \mu(p_\text{gen})$")
-            ax.set_ylabel(r"$w_\text{disc}$")
+            ax.set_xlabel(r"$\sigma(p_\text{model}) / \mu(p_\text{model})$")
+            ax.set_ylabel(r"$w$")
 
             plt.savefig(pdf, format="pdf")
             plt.close()
@@ -430,9 +430,9 @@ def plot_bgen_weights(self, file: str):
             )
             #cb = plt.colorbar(img)
             #cb.set_label('norm.')
-            axs[0].set_ylabel(r"$\sigma(p_\text{gen}) / \mu(p_\text{gen})$")
+            axs[0].set_ylabel(r"$\sigma(p_\text{model}) / \mu(p_\text{model})$")
             axs[1].set_ylabel(r"median $\sigma / \mu$")
-            axs[1].set_xlabel(r"$w_\text{disc}$")
+            axs[1].set_xlabel(r"$w$")
             fig.align_ylabels()
             self.corner_text(axs[1], self.title, "right", "top", h_offset=0.15)
             plt.savefig(pdf, format="pdf")
@@ -461,7 +461,7 @@ def plot_bgen_weights(self, file: str):
                 label="Fit",
                 color=self.colors[1]
             )
-            ax.set_xlabel(r"$\mu(p_\text{gen}) (w_\text{disc} - 1) / \sigma(p_\text{gen})$")
+            ax.set_xlabel(r"$\mu(p_\text{model}) (w - 1) / \sigma(p_\text{model})$")
             ax.set_ylabel("normalized")
             #ax.set_yscale("log")
             ax.set_xlim(bins[0], bins[-1])
@@ -499,7 +499,7 @@ def plot_bgen_weights(self, file: str):
                 label="Fit",
                 color=self.colors[1]
             )
-            ax.set_xlabel(r"$\log w_\text{disc} / \sigma(\log p_\text{gen})$")
+            ax.set_xlabel(r"$\log w / \sigma(\log p_\text{model})$")
             ax.set_ylabel("normalized")
             #ax.set_yscale("log")
             ax.set_xlim(bins[0], bins[-1])
@@ -524,8 +524,8 @@ def plot_bgen_weights(self, file: str):
                 label = "",
                 color = self.colors[1]
             )
-            ax.set_xlabel(r"$w_\text{disc}$")
-            ax.set_ylabel(r"median $\sigma(p_\text{gen}) / \mu(p_\text{gen})$")
+            ax.set_xlabel(r"$w$")
+            ax.set_ylabel(r"median $\sigma(p_\text{model}) / \mu(p_\text{model})$")
             self.corner_text(ax, self.title, "right", "top")
             plt.savefig(pdf, format="pdf")
             plt.close()