Update plotter.py

* minor update for 1d plots
mathLab · Nov 17, 2023 · 48b2e33 · 48b2e33
1 parent 9bf057e
commit 48b2e33
Showing 1 changed file with 1 addition and 2 deletions.
diff --git 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,