add bbox_inches tight to savefig commands

SMTG-Bham · Mar 19, 2021 · a69a005 · a69a005
1 parent 55e4010
commit a69a005
Showing 1 changed file with 2 additions and 2 deletions.
diff --git a/surfaxe/io.py b/surfaxe/io.py
@@ -224,7 +224,7 @@ def plot_bond_analysis(bond, df=None, filename=None, width=6, height=5, dpi=300,
     ax.set_ylabel("Bond distance / Å")
     ax.legend(['{}-{} bond'.format(bond[0], bond[1])])
     plt.xlabel("Fractional coordinate in c")
-    fig.savefig(plt_fname, facecolor='w')
+    fig.savefig(plt_fname, facecolor='w', bbox_inches='tight')
 
 
 def plot_electrostatic_potential(df=None, filename=None, dpi=300, width=6,
@@ -268,7 +268,7 @@ def plot_electrostatic_potential(df=None, filename=None, dpi=300, width=6,
     ax.axes.xaxis.set_visible(False)
     ax.legend()
     plt.ylabel('Potential / eV')
-    fig.savefig(plt_fname, facecolor='w')
+    fig.savefig(plt_fname, facecolor='w', bbox_inches='tight')
 
 def plot_surfen(df, joules=True, time_taken=True, colors=None, dpi=300, width=6,
 height=5, heatmap=False, cmap='Wistia',  plt_fname='surface_energy.png'):