Merge pull request #36 from mj-will/fix-plots-not-closing

Close figures that weren't closed

nessai/nestedsampler.py

@@ -945,8 +945,8 @@ def plot_state(self):
 
         fig.tight_layout()
         fig.subplots_adjust(top=0.95)
-
         fig.savefig(f'{self.output}/state.png')
+        plt.close(fig)
 
     def plot_trace(self):
         """

nessai/plot.py

@@ -146,7 +146,7 @@ def plot_1d_comparison(*live_points, parameters=None, labels=None,
     plt.tight_layout()
     if filename is not None:
         fig.savefig(filename, bbox_inches='tight')
-        plt.close()
+        plt.close(fig)
     else:
         return fig
 
@@ -207,7 +207,7 @@ def plot_indices(indices, nlive=None, filename=None, plot_breakdown=True):
 
     if filename is not None:
         plt.savefig(filename, bbox_inches='tight')
-        plt.close()
+        plt.close(fig)
     else:
         return fig
 
@@ -241,7 +241,7 @@ def plot_loss(epoch, history, filename=None):
 
     if filename is not None:
         plt.savefig(filename, bbox_inches='tight')
-        plt.close()
+        plt.close(fig)
     else:
         return fig
 

nessai/proposal/flowproposal.py

@@ -905,6 +905,7 @@ def train(self, x, plot=True):
             plt.hist(np.sqrt(np.sum(z_training_data ** 2, axis=1)), 'auto')
             plt.xlabel('Radius')
             fig.savefig(block_output + 'radial_dist.png')
+            plt.close(fig)
 
             plot_1d_comparison(z_training_data, z_gen,
                                labels=['z_live_points', 'z_generated'],
@@ -1422,6 +1423,7 @@ def plot_pool(self, z, x):
             plt.tight_layout()
             fig.savefig(
                 f'{self.output}/pool_{self.populated_count}_log_q.png')
+            plt.close(fig)
 
     def resume(self, model, flow_config, weights_file=None):
         """