fix(bo): Off-by-1 in range for generating new config_ids

automl · Jan 26, 2025 · 3d98089 · 3d98089
1 parent 7755a73
commit 3d98089
Showing 1 changed file with 3 additions and 1 deletion.
diff --git a/neps/optimizers/bayesian_optimization.py b/neps/optimizers/bayesian_optimization.py
@@ -89,7 +89,9 @@ def __call__(
 
         n_to_sample = 1 if n is None else n
         n_sampled = len(trials)
-        config_ids = iter(str(i + 1) for i in range(n_sampled, n_sampled + n_to_sample))
+        config_ids = iter(
+            str(i + 1) for i in range(n_sampled, n_sampled + n_to_sample + 1)
+        )
 
         # If the amount of configs evaluated is less than the initial design
         # requirement, keep drawing from initial design