Add information on handling of stopIteration error

automl · Jul 30, 2024 · fa96bee · fa96bee
1 parent 7058b45
commit fa96bee
Show file tree

Hide file tree

Showing 3 changed files with 13 additions and 0 deletions.
diff --git a/CHANGELOG.md b/CHANGELOG.md
@@ -1,3 +1,8 @@
+# 2.2.1
+
+## Improvements
+- Add logger information on handling of stopIteration error (#960)
+
 # 2.2.0
 
 ## Features

diff --git a/smac/intensifier/intensifier.py b/smac/intensifier/intensifier.py
@@ -232,6 +232,10 @@ def __iter__(self) -> Iterator[TrialInfo]:
                     fails = -1
                 except StopIteration:
                     # We stop if we don't find any configuration anymore
+                    logger.warning(
+                        "If you assume your configspace was not yet exhausted, try to "
+                        "increase the number of retries in the config selector."
+                    )
                     return
             else:
                 logger.debug("Start finding a new challenger in the queue:")

diff --git a/smac/intensifier/successive_halving.py b/smac/intensifier/successive_halving.py
@@ -448,6 +448,10 @@ def __iter__(self) -> Iterator[TrialInfo]:  # noqa: D102
                     configs.append(config)
                 except StopIteration:
                     # We stop if we don't find any configuration anymore
+                    logger.warning(
+                        "If you assume your configspace was not yet exhausted, try to "
+                        "increase the number of retries in the config selector."
+                    )
                     return
 
             # We keep track of the seed so we always evaluate on the same instances