Skip to content

Commit

Permalink
Move objective statement
Browse files Browse the repository at this point in the history
qNegIntegratedPosteriorVariance does not accept an objective
  • Loading branch information
AdrianSosic committed Jan 22, 2025
1 parent 4fdd687 commit ab0da62
Showing 1 changed file with 3 additions and 3 deletions.
6 changes: 3 additions & 3 deletions baybe/acquisition/base.py
Original file line number Diff line number Diff line change
Expand Up @@ -116,13 +116,13 @@ def to_botorch(

if issubclass(acqf_cls, bo_acqf.AnalyticAcquisitionFunction):
additional_params["maximize"] = False
elif issubclass(acqf_cls, bo_acqf.qNegIntegratedPosteriorVariance):
# qNIPV is valid but does not require any adjusted params
pass
elif self.is_mc:
additional_params["objective"] = LinearMCObjective(
torch.tensor([-1.0])
)
elif issubclass(acqf_cls, bo_acqf.qNegIntegratedPosteriorVariance):
# qNIPV is valid but does not require any adjusted params
pass
else:
raise ValueError(
f"Unsupported acquisition function type: {acqf_cls}."
Expand Down

0 comments on commit ab0da62

Please sign in to comment.