File tree Expand file tree Collapse file tree 1 file changed +3
-3
lines changed Expand file tree Collapse file tree 1 file changed +3
-3
lines changed Original file line number Diff line number Diff line change @@ -517,10 +517,10 @@ def test_continuous_step(self):
517
517
X [b_i , cont_dims ] = torch .rand (d_cont , device = self .device )
518
518
519
519
best_f = model (X ).min ()
520
- ei = PosteriorMean (model )
520
+ mean_as_acq = PosteriorMean (model )
521
521
X_new , ei_val = continuous_step (
522
522
opt_inputs = _make_opt_inputs (
523
- acq_function = ei ,
523
+ acq_function = mean_as_acq ,
524
524
bounds = bounds ,
525
525
options = {"maxiter_continuous" : 32 },
526
526
return_best_only = False ,
@@ -542,7 +542,7 @@ def test_continuous_step(self):
542
542
X_ [:, :2 ] = 1.0 # To satisfy the constraint.
543
543
X_new , ei_val = continuous_step (
544
544
opt_inputs = _make_opt_inputs (
545
- acq_function = ei ,
545
+ acq_function = mean_as_acq ,
546
546
bounds = bounds ,
547
547
options = {"maxiter_continuous" : 32 },
548
548
fixed_features = {fixed_binary : 1 , fixed_cont : 0.5 },
You can’t perform that action at this time.
0 commit comments