From 94e5dfac77c80f2de9d30fbfb6dd08ed4f45c5f0 Mon Sep 17 00:00:00 2001
From: xuyuon <116078673+xuyuon@users.noreply.github.com>
Date: Thu, 25 Jul 2024 12:05:53 -0400
Subject: [PATCH] Updated runManager.py

---
 src/jimgw/single_event/runManager.py | 6 +++++-
 1 file changed, 5 insertions(+), 1 deletion(-)

diff --git a/src/jimgw/single_event/runManager.py b/src/jimgw/single_event/runManager.py
index 66790c1d..6a3b3091 100644
--- a/src/jimgw/single_event/runManager.py
+++ b/src/jimgw/single_event/runManager.py
@@ -398,27 +398,31 @@ def plot_diagnostic(self, path: str = "diagnostic.jpeg", **kwargs):
         chains, log_prob, local_accs, global_accs, loss_vals = summary.values()
         log_prob = np.array(log_prob)
 
-        plt.figure(figsize=(6, 6))
+        plt.figure(figsize=(10, 10))
         axs = [plt.subplot(2, 2, i + 1) for i in range(4)]
         plt.sca(axs[0])
         plt.title("log probability")
         plt.plot(log_prob.mean(0))
         plt.xlabel("iteration")
+        plt.xlim(0, None)
 
         plt.sca(axs[1])
         plt.title("NF loss")
         plt.plot(loss_vals.reshape(-1))
         plt.xlabel("iteration")
+        plt.xlim(0, None)
 
         plt.sca(axs[2])
         plt.title("Local Acceptance")
         plt.plot(local_accs.mean(0))
         plt.xlabel("iteration")
+        plt.xlim(0, None)
 
         plt.sca(axs[3])
         plt.title("Global Acceptance")
         plt.plot(global_accs.mean(0))
         plt.xlabel("iteration")
+        plt.xlim(0, None)
         plt.tight_layout()
 
         plt.savefig(path)