From 0b80af2fa46d03d804f3e6871fc38965e879728b Mon Sep 17 00:00:00 2001
From: Farid Mohammadi <farid.mohammadi@iws.uni-stuttgart.de>
Date: Fri, 3 Jun 2022 12:12:48 +0200
Subject: [PATCH] [bayes_inference][bayes_model_comparison] fix small bugs.

---
 .gitignore                                                  | 2 ++
 src/bayesvalidrox/bayes_inference/bayes_inference.py        | 1 +
 src/bayesvalidrox/bayes_inference/bayes_model_comparison.py | 4 +++-
 3 files changed, 6 insertions(+), 1 deletion(-)

diff --git a/.gitignore b/.gitignore
index 61449c637..423a52d16 100644
--- a/.gitignore
+++ b/.gitignore
@@ -4,11 +4,13 @@
 *.png
 !docs/logo/*.png
 *.svg
+!examples/model-comparison/reference/*.svg
 *.zip
 *.vtu
 *.vtp
 *.pvd
 *.pkl
+!examples/model-comparison/reference/*.pkl
 
 # Ignore __pycache__ directories
 */__pycache__/
diff --git a/src/bayesvalidrox/bayes_inference/bayes_inference.py b/src/bayesvalidrox/bayes_inference/bayes_inference.py
index cb7d871ec..714d10a60 100644
--- a/src/bayesvalidrox/bayes_inference/bayes_inference.py
+++ b/src/bayesvalidrox/bayes_inference/bayes_inference.py
@@ -524,6 +524,7 @@ class BayesInference:
                 for xi in range(yi):
                     ax = axes[yi, xi]
                     ax.set_xlim(MetaModel.bound_tuples[xi])
+        plt.close()
 
         # Turn off gridlines
         for ax in figPosterior.axes:
diff --git a/src/bayesvalidrox/bayes_inference/bayes_model_comparison.py b/src/bayesvalidrox/bayes_inference/bayes_model_comparison.py
index 7ce17841e..e054d23ef 100644
--- a/src/bayesvalidrox/bayes_inference/bayes_model_comparison.py
+++ b/src/bayesvalidrox/bayes_inference/bayes_model_comparison.py
@@ -36,6 +36,7 @@ class BayesModelComparison:
 
         # Compute total number of the measurement points
         MetaModel = list(modelDict.items())[0][1]
+        MetaModel.ModelObj.read_observation()
         self.n_meas = MetaModel.ModelObj.n_obs
 
         # Generate justifiability data
@@ -168,9 +169,10 @@ class BayesModelComparison:
                          ms=10, linewidth=2, label=model
                          )
 
-            plt.title(name.replace('_', '$-$'))
+            plt.title(f"Data generated by: {name.replace('_', '$-$')}")
             plt.ylabel("Weights")
             plt.xlabel("No. of measurement points")
+            ax.set_xticks(list(range(1, self.n_meas+1)))
             plt.legend(loc="best")
             fig.savefig(
                 f'{directory}modelWeights_{name}.svg', bbox_inches='tight'
-- 
GitLab