feat: unify optimize and optimize_with via blanket Objective impl

- Add blanket `impl Objective<V> for Fn(&mut Trial) -> Result<V, E>`
  so closures work directly with `optimize`
- Rewrite optimize, optimize_async, optimize_parallel to accept
  `impl Objective<V>` with before_trial/after_trial hooks
- Remove optimize_with, optimize_with_async, optimize_with_parallel
- Remove max_retries and retry logic from Objective trait
- Add explicit closure type annotations for HRTB inference
- Convert FnMut test closures to Fn via RefCell/Cell
This commit is contained in:
Manuel Raimann
2026-02-12 13:09:14 +01:00
parent c20a53dfba
commit 47b5f9cec8
41 changed files with 316 additions and 793 deletions
+3 -3
View File
@@ -27,7 +27,7 @@ fn bohb_converges_on_quadratic() {
let x_param = FloatParam::new(-10.0, 10.0);
study
.optimize(60, |trial| {
.optimize(60, |trial: &mut optimizer::Trial| {
let x = x_param.suggest(trial)?;
// Report intermediate values at budget steps 1, 3, 9
@@ -66,7 +66,7 @@ fn bohb_with_pruning() {
let x_param = FloatParam::new(-5.0, 5.0);
study
.optimize(40, |trial| {
.optimize(40, |trial: &mut optimizer::Trial| {
let x = x_param.suggest(trial)?;
let obj = x * x;
@@ -112,7 +112,7 @@ fn bohb_uses_budget_conditioned_history() {
let x_param = FloatParam::new(0.0, 10.0);
study
.optimize(30, |trial| {
.optimize(30, |trial: &mut optimizer::Trial| {
let x = x_param.suggest(trial)?;
// Intermediate values that guide optimization toward x=2
trial.report(1, (x - 2.0).powi(2) + 1.0);