Add check for None

automl · Jan 15, 2025 · 9bbb9e3 · 9bbb9e3
1 parent 37701e3
commit 9bbb9e3
Showing 1 changed file with 1 addition and 1 deletion.
diff --git a/smac/acquisition/maximizer/local_search.py b/smac/acquisition/maximizer/local_search.py
@@ -338,7 +338,7 @@ def _search(
         while np.any(active):
 
             # If the maximum number of steps is reached, stop the local search
-            if num_iters == self._max_steps:
+            if num_iters is not None and num_iters == self._max_steps:
                 break
 
             num_iters += 1