From e22c675cf979324af1f6a2ddf27b086f7dfd5b17 Mon Sep 17 00:00:00 2001
From: hsulab <ahcigar@foxmail.com>
Date: Fri, 5 Jul 2024 14:57:29 -0400
Subject: [PATCH] update plotting style

---
 src/gdpx/validator/diffusion_coefficient.py | 6 +++---
 1 file changed, 3 insertions(+), 3 deletions(-)

diff --git a/src/gdpx/validator/diffusion_coefficient.py b/src/gdpx/validator/diffusion_coefficient.py
index 95d8c874..1f2e8511 100644
--- a/src/gdpx/validator/diffusion_coefficient.py
+++ b/src/gdpx/validator/diffusion_coefficient.py
@@ -58,9 +58,9 @@ def plot_msd(wdir, names, lagtimes, timeseries, start_step=20, end_step=60, pref
     ax.set_ylabel("MSD [Å^2]")
     ax.set_xlabel("Time [ps]")
 
-    ax.legend()
+    ax.legend(fontsize=12)
 
-    plt.savefig(wdir/ f"{prefix}msd.png")
+    fig.savefig(wdir/ f"{prefix}msd.png", bbox_inches="tight")
 
     return
 
@@ -207,4 +207,4 @@ def _irun(self, data, prefix="", labels=None):
 
 
 if __name__ == "__main__":
-    ...
\ No newline at end of file
+    ...