Skip to content

Commit

Permalink
Don't allow unused **kwargs in input_constructors except for a define…
Browse files Browse the repository at this point in the history
…d set of exceptions (#1772)

Summary:
Pull Request resolved: #1772

X-link: pytorch/botorch#1872

[x] Remove unused arguments from input constructors and related functions. The idea is especially not to let unused keyword arguments disappear into `**kwargs` and be silently ignored
[x] add arguments to some input constructors so they don't need any `**kwargs`
[x] Add a decorator that ensures that each input constructor can accept a certain set of keyword arguments, even if those are not used are the constructor, while still erroring on
[ ] Prevent arguments from having different defaults in the input constructors as in acquisition functions

Reviewed By: SebastianAment

Differential Revision: D46519588

fbshipit-source-id: 44c5c6e99a4e3ae3b287da9901a9e20af322239e
  • Loading branch information
esantorella authored and facebook-github-bot committed Aug 9, 2023
1 parent ecbb88f commit 6ed7963
Showing 1 changed file with 4 additions and 2 deletions.
6 changes: 4 additions & 2 deletions ax/models/torch/botorch_modular/sebo.py
Original file line number Diff line number Diff line change
Expand Up @@ -61,8 +61,8 @@ def __init__(

tkwargs = {"dtype": surrogate.dtype, "device": surrogate.device}
options = options or {}
self.penalty_name: str = options.get("penalty", "L0_norm")
self.target_point: Tensor = options.get("target_point", None)
self.penalty_name: str = options.pop("penalty", "L0_norm")
self.target_point: Tensor = options.pop("target_point", None)
if self.target_point is None:
raise ValueError("please provide target point.")
self.target_point.to(**tkwargs) # pyre-ignore
Expand Down Expand Up @@ -93,6 +93,8 @@ def __init__(
)

# instantiate botorch_acqf_class
if not issubclass(botorch_acqf_class, qExpectedHypervolumeImprovement):
raise ValueError("botorch_acqf_class must be qEHVI to use SEBO")
super().__init__(
surrogates={"sebo": surrogate_f},
search_space_digest=search_space_digest,
Expand Down

0 comments on commit 6ed7963

Please sign in to comment.