From f9bf55b9c7d91daecd18f9268d67083bae4a77d1 Mon Sep 17 00:00:00 2001
From: Colm Talbot <colm.talbot@ligo.org>
Date: Fri, 12 Aug 2022 14:31:23 +0000
Subject: [PATCH] ENH: Fix evidence labels for multi-result plotting
---
bilby/core/result.py | 15 ++++++++++-----
1 file changed, 10 insertions(+), 5 deletions(-)
diff --git a/bilby/core/result.py b/bilby/core/result.py
index c75e1d57d..c5475c98e 100644
--- a/bilby/core/result.py
+++ b/bilby/core/result.py
@@ -1939,12 +1939,17 @@ def plot_multiple(results, filename=None, labels=None, colours=None,
if evidences:
if np.isnan(results[0].log_bayes_factor):
- template = r' $\mathrm{{ln}}(Z)={lnz:1.3g}$'
+ template = r'{label} $\mathrm{{ln}}(Z)={lnz:1.3g}$'
else:
- template = r' $\mathrm{{ln}}(B)={lnbf:1.3g}$'
- labels = [template.format(lnz=result.log_evidence,
- lnbf=result.log_bayes_factor)
- for ii, result in enumerate(results)]
+ template = r'{label} $\mathrm{{ln}}(B)={lnbf:1.3g}$'
+ labels = [
+ template.format(
+ label=label,
+ lnz=result.log_evidence,
+ lnbf=result.log_bayes_factor,
+ )
+ for label, result in zip(labels, results)
+ ]
axes = fig.get_axes()
ndim = int(np.sqrt(len(axes)))
--
GitLab