Skip to content

Commit f23e90b

Browse files
committed
Drop output dimension of acquisition function values
1 parent 8c73830 commit f23e90b

File tree

1 file changed

+1
-1
lines changed

1 file changed

+1
-1
lines changed

botorch/optim/optimize.py

Lines changed: 1 addition & 1 deletion
Original file line numberDiff line numberDiff line change
@@ -192,7 +192,7 @@ def _optimize_acqf_all_features_fixed(
192192
X = X.expand(q, *X.shape)
193193
with torch.no_grad():
194194
acq_value = acq_function(X)
195-
return X, acq_value
195+
return X, acq_value[0]
196196

197197

198198
def _validate_sequential_inputs(opt_inputs: OptimizeAcqfInputs) -> None:

0 commit comments

Comments
 (0)