0.15.24

save_experiment has arg overwrite

Author: bartzbeielstein

pyproject.toml

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

src/spotpython/spot/spot.py

@@ -2295,9 +2295,18 @@ def parallel_plot(self, show=False) -> go.Figure:
             fig.show()
         return fig
 
-    def save_experiment(self, filename=None) -> None:
+    def save_experiment(self, filename=None, overwrite=True) -> None:
         """
         Save the experiment to a file.
+
+        Args:
+            filename (str):
+                The filename of the experiment file.
+            overwrite (bool):
+                If `True`, the file will be overwritten if it already exists. Default is `True`.
+
+        Returns:
+            None
         """
         # Remove or close any unpickleable objects, e.g., the spot_writer
         self.close_and_del_spot_writer()
@@ -2333,6 +2342,11 @@ def save_experiment(self, filename=None) -> None:
         if filename is None and PREFIX is not None:
             filename = get_experiment_filename(PREFIX)
 
+        # Check if the file already exists
+        if filename is not None and os.path.exists(filename) and not overwrite:
+            print(f"Error: File {filename} already exists. Use overwrite=True to overwrite the file.")
+            return
+
         # Serialize the experiment dictionary to the pickle file
         if filename is not None:
             with open(filename, "wb") as handle: