update plotting style

hsulab · Jul 9, 2024 · e22c675 · e22c675
1 parent b6e5b97
commit e22c675
Showing 1 changed file with 3 additions and 3 deletions.
diff --git 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__":
-    ...
+    ...