fix: Ensure we do the last intial design

automl · Aug 28, 2024 · b135c74 · b135c74
1 parent 8658d70
commit b135c74
Showing 1 changed file with 1 addition and 1 deletion.
diff --git a/neps/optimizers/bayesian_optimization/optimizer.py b/neps/optimizers/bayesian_optimization/optimizer.py
@@ -202,7 +202,7 @@ def ask(
             self.initial_design_.extend(configs)
 
         # If we havn't passed the intial design phase
-        if len(trials) < len(self.initial_design_):
+        if len(trials) <= len(self.initial_design_):
             config = self.initial_design_[len(trials) - 1]
             sample = SampledConfig(id=config_id, config=config, previous_config_id=None)
             return sample, optimizer_state