Add fix

automl · Jan 15, 2025 · 1a1b9f8 · 1a1b9f8
1 parent 2659f4f
commit 1a1b9f8
Showing 1 changed file with 5 additions and 0 deletions.
diff --git a/smac/acquisition/maximizer/local_search.py b/smac/acquisition/maximizer/local_search.py
@@ -336,6 +336,11 @@ def _search(
 
         num_iters = 0
         while np.any(active):
+
+            # If the maximum number of steps is reached, stop the local search
+            if num_iters == self._max_steps:
+                break
+
             num_iters += 1
             # Whether the i-th local search improved. When a new neighborhood is generated, this is used to determine
             # whether a step was made (improvement) or not (iterator exhausted)