0.31.15

Kriging plot adds points

Author: bartzbeielstein

pyproject.toml

@@ -7,7 +7,7 @@ build-backend = "setuptools.build_meta"
 
 [project]
 name = "spotpython"
-version = "0.31.14"
+version = "0.31.15"
 authors = [
   { name="T. Bartz-Beielstein", email="tbb@bartzundbartz.de" }
 ]

src/spotpython/surrogate/kriging.py

@@ -731,7 +731,7 @@ def objective(logtheta_lambda):
         result = differential_evolution(objective, bounds)
         return result.x, result.fun
 
-    def plot(self, i: int = 0, j: int = 1, show: Optional[bool] = True) -> None:
+    def plot(self, i: int = 0, j: int = 1, show: Optional[bool] = True, add_points: bool = True) -> None:
         """
         This function plots 1D and 2D surrogates.
         Only for compatibility with the old Kriging implementation.
@@ -746,6 +746,8 @@ def plot(self, i: int = 0, j: int = 1, show: Optional[bool] = True) -> None:
             show (bool):
                 If `True`, the plots are displayed.
                 If `False`, `plt.show()` should be called outside this function.
+            add_points (bool):
+                If `True`, the points from the design are added to the plot.
 
         Returns:
             None
@@ -795,4 +797,4 @@ def plot(self, i: int = 0, j: int = 1, show: Optional[bool] = True) -> None:
             if show:
                 plt.show()
         else:
-            plotkd(model=self, X=self.X_, y=self.y_, i=i, j=j, show=show, var_type=self.var_type)
+            plotkd(model=self, X=self.X_, y=self.y_, i=i, j=j, show=show, var_type=self.var_type, add_points=True)