From 1094cf7ad6bd505ca5d51f1621d5716344865bf1 Mon Sep 17 00:00:00 2001
From: Ekaterina Kazmina <ekaterinakazmina88@gmail.com>
Date: Wed, 3 Jun 2026 15:59:48 +0000
Subject: [PATCH 1/2] fix: remove plt.close() from show(), expose get_figure()
 and get_axes()

Fixes #771.

Previously, calling .show() would call plt.close() immediately after
plt.show(), destroying the figure and making it impossible to:
  - further customise the plot (add titles, annotations, change limits)
  - save it again with different settings
  - access the figure/axes objects for downstream use

Changes:
  - Remove the bare plt.close() call from Plot.show()
  - Add Plot.get_figure() -> matplotlib.figure.Figure
  - Add Plot.get_axes()   -> matplotlib.axes.Axes (or ndarray)
  - Clarify close_on_destroy comment: users can set it to False to keep
    the figure alive after the Plot object is garbage-collected

Users who relied on plt.close() being called automatically can close
explicitly after they are done:
    plot.show()
    plt.close(plot.fig)
---
 pymoo/core/plot.py | 42 ++++++++++++++++++++++++++++++++++++++++--
 1 file changed, 40 insertions(+), 2 deletions(-)

diff --git a/pymoo/core/plot.py b/pymoo/core/plot.py
index e9e7cfea..cbe3147d 100644
--- a/pymoo/core/plot.py
+++ b/pymoo/core/plot.py
@@ -31,12 +31,15 @@ def __init__(self,
         # change the font of plots to serif (looks better)
         plt.rc('font', family='serif')
 
-        # the matplotlib classes
+        # the matplotlib classes — publicly accessible after do()/show()/save()
+        # you can also use get_figure() and get_axes() helper methods
         self.fig = fig
         self.ax = ax
         self.figsize = figsize
 
         # whether the figure should be closed when object is destroyed
+        # set to False if you want to keep the figure alive after the Plot object
+        # goes out of scope (e.g. to further customise or display it)
         self.close_on_destroy = close_on_destroy
 
         # the title of the figure - can also be a list if subfigures
@@ -172,10 +175,45 @@ def show(self, **kwargs):
         # in a notebook the plot method need not to be called explicitly
         if not in_notebook() and matplotlib.get_backend().lower() != "agg":
             plt.show(**kwargs)
-            plt.close()
+            # NOTE: plt.close() removed — callers can now close the figure themselves
+            # if needed, e.g.: plot.fig.savefig(...); plt.close(plot.fig)
 
         return self
 
+    def get_figure(self):
+        """Return the underlying matplotlib Figure object.
+
+        Useful for further customisation after calling .show() or .save()::
+
+            plot = Scatter().add(F).show()
+            fig = plot.get_figure()
+            fig.suptitle("My custom title")
+            fig.savefig("result.png")
+
+        Returns
+        -------
+        matplotlib.figure.Figure
+        """
+        self.plot_if_not_done_yet()
+        return self.fig
+
+    def get_axes(self):
+        """Return the matplotlib Axes object (or array of Axes for multi-panel plots).
+
+        Useful for adding annotations, changing limits, or any other per-axis
+        customisation after plotting::
+
+            plot = Scatter().add(F).show()
+            ax = plot.get_axes()
+            ax.set_xlim(0, 1)
+
+        Returns
+        -------
+        matplotlib.axes.Axes or numpy.ndarray of Axes
+        """
+        self.plot_if_not_done_yet()
+        return self.ax
+
     def save(self, fname, **kwargs):
         self.plot_if_not_done_yet()
         set_if_none(kwargs, "bbox_inches", "tight")

From 934a7c84a9162f5718f71763dafb70035273f185 Mon Sep 17 00:00:00 2001
From: katerinakzmn <151528753+katerinakzmn@users.noreply.github.com>
Date: Thu, 4 Jun 2026 14:08:07 +0300
Subject: [PATCH 2/2] Update plot.py

---
 pymoo/core/plot.py | 26 --------------------------
 1 file changed, 26 deletions(-)

diff --git a/pymoo/core/plot.py b/pymoo/core/plot.py
index cbe3147d..e782f1cf 100644
--- a/pymoo/core/plot.py
+++ b/pymoo/core/plot.py
@@ -181,36 +181,10 @@ def show(self, **kwargs):
         return self
 
     def get_figure(self):
-        """Return the underlying matplotlib Figure object.
-
-        Useful for further customisation after calling .show() or .save()::
-
-            plot = Scatter().add(F).show()
-            fig = plot.get_figure()
-            fig.suptitle("My custom title")
-            fig.savefig("result.png")
-
-        Returns
-        -------
-        matplotlib.figure.Figure
-        """
         self.plot_if_not_done_yet()
         return self.fig
 
     def get_axes(self):
-        """Return the matplotlib Axes object (or array of Axes for multi-panel plots).
-
-        Useful for adding annotations, changing limits, or any other per-axis
-        customisation after plotting::
-
-            plot = Scatter().add(F).show()
-            ax = plot.get_axes()
-            ax.set_xlim(0, 1)
-
-        Returns
-        -------
-        matplotlib.axes.Axes or numpy.ndarray of Axes
-        """
         self.plot_if_not_done_yet()
         return self.ax