Save after ask

hadarshavit · Oct 12, 2024 · fdbe11e · fdbe11e
1 parent 81b2380
commit fdbe11e
Show file tree

Hide file tree

Showing 2 changed files with 3 additions and 1 deletion.
diff --git a/smac/__init__.py b/smac/__init__.py
@@ -19,7 +19,7 @@
     Copyright {datetime.date.today().strftime('%Y')}, Marius Lindauer, Katharina Eggensperger,
     Matthias Feurer, André Biedenkapp, Difan Deng, Carolin Benjamins, Tim Ruhkopf, René Sass
     and Frank Hutter"""
-version = "2.2.0post10"
+version = "2.2.0post11"
 
 
 try:

diff --git a/smac/main/smbo.py b/smac/main/smbo.py
@@ -303,6 +303,8 @@ def optimize(self, *, data_to_scatter: dict[str, Any] | None = None) -> Configur
                 # We submit the trial to the runner
                 # In multi-worker mode, SMAC waits till a new worker is available here
                 self._runner.submit_trial(trial_info=trial_info, **dask_data_to_scatter)
+
+                self.save()
             except StopIteration:
                 self._stop = True