File tree Expand file tree Collapse file tree 2 files changed +4
-2
lines changed Expand file tree Collapse file tree 2 files changed +4
-2
lines changed Original file line number Diff line number Diff line change @@ -273,7 +273,7 @@ def __init__(
273
273
constraints : list [Callable [[Tensor ], Tensor ]] | None = None ,
274
274
eta : Tensor | float = 1e-3 ,
275
275
fat : bool = True ,
276
- prune_baseline : bool = False ,
276
+ prune_baseline : bool = True ,
277
277
cache_root : bool = True ,
278
278
tau_max : float = TAU_MAX ,
279
279
tau_relu : float = TAU_RELU ,
Original file line number Diff line number Diff line change @@ -479,7 +479,9 @@ def test_optimize_acqf_mixed_binary_only(self) -> None:
479
479
# get multiple candidates
480
480
root = torch .zeros (dim , device = self .device )
481
481
model = QuadraticDeterministicModel (root )
482
- acqf = qLogNoisyExpectedImprovement (model = model , X_baseline = train_X )
482
+ acqf = qLogNoisyExpectedImprovement (
483
+ model = model , X_baseline = train_X , prune_baseline = False
484
+ )
483
485
options ["initialization_strategy" ] = "equally_spaced"
484
486
candidates , _ = optimize_acqf_mixed_alternating (
485
487
acq_function = acqf ,
You can’t perform that action at this time.
0 commit comments