Bugfix: Plotting xlim / ylim

shirtsgroup · Jan 30, 2018 · 28a1b2a · 28a1b2a
1 parent ce0589d
commit 28a1b2a
Showing 1 changed file with 3 additions and 1 deletion.
diff --git a/physical_validation/util/plot.py b/physical_validation/util/plot.py
@@ -89,7 +89,7 @@ def plot(res, legend=None, title=None,
     ax.axis('auto')
     if xlim is not None:
         ax.set_xlim(xlim)
-    else:
+    elif np.isfinite(xmin) and np.isfinite(xmax):
         ax.set_xlim([xmin, xmax])
     if ylim is not None:
         ax.set_ylim(ylim)
@@ -127,3 +127,5 @@ def plot(res, legend=None, title=None,
         fig.savefig(filename + '.pdf', dpi=300)
     if screen:
         fig.show()
+
+    plt.close(fig)