From 48b2e339b5e789ab41c09eed84e4eadd721b3e28 Mon Sep 17 00:00:00 2001
From: Dario Coscia <93731561+dario-coscia@users.noreply.github.com>
Date: Thu, 16 Nov 2023 17:47:26 +0100
Subject: [PATCH] Update plotter.py

* minor update for 1d plots
---
 pina/plotter.py | 3 +--
 1 file changed, 1 insertion(+), 2 deletions(-)

diff --git a/pina/plotter.py b/pina/plotter.py
index cd3a0b7f..7eebf63c 100644
--- a/pina/plotter.py
+++ b/pina/plotter.py
@@ -197,7 +197,7 @@ def plot(self,
         truth_solution = getattr(solver.problem, 'truth_solution', None)
 
         if len(v) == 1:
-            self._1d_plot(pts, predicted_output, method, truth_solution,
+            self._1d_plot(pts.extract(v), predicted_output, method, truth_solution,
                           **kwargs)
         elif len(v) == 2:
             self._2d_plot(pts, predicted_output, v, res, method, truth_solution,
@@ -208,7 +208,6 @@ def plot(self,
             plt.savefig(filename)
         else:
             plt.show()
-        plt.close()
 
     def plot_loss(self,
                   trainer,