close figure after it is saved

Without plt.close(), the figure is kept open and wastes memory.
drorlab · Aug 13, 2021 · bdb833f · bdb833f
1 parent 60facb5
commit bdb833f
Showing 1 changed file with 1 addition and 0 deletions.
diff --git a/pensa/statesinfo/discrete_states.py b/pensa/statesinfo/discrete_states.py
@@ -439,6 +439,7 @@ def get_intersects(gaussians, distribution, Gauss_xvals, write_plots=None,write_
         plt.title(write_name)        
         plt.ioff()
         plt.savefig('ssi_plots/'+write_name+".png")
+        plt.close()
 
     return all_intersects