More cleanup of memory maps

Author: matteobachetti

hendrics/efsearch.py

@@ -755,12 +755,15 @@ def plot_transient_search(results, gif_name=None):
             else:
                 axf.set_xlim([xmin, xmax])
 
-            fig.canvas.draw()
-            image = np.frombuffer(fig.canvas.buffer_rgba().cast("B"), dtype="uint8")
-            image = image.reshape(fig.canvas.get_width_height()[::-1] + (4,))
+        fig.canvas.draw()
+        image = np.frombuffer(fig.canvas.buffer_rgba().cast("B"), dtype="uint8")
+        image = image.reshape(fig.canvas.get_width_height()[::-1] + (4,))
 
         plt.close(fig)
         all_images.append(image)
+
+    if hasattr(results.stats, "filename"):
+        os.remove(results.stats.filename)
     vstack(max_stats_rows).write(result_name, overwrite=True)
 
     imageio.v3.imwrite(gif_name, all_images, duration=1000.0)