Skip to content

Commit

Permalink
Add **kwargs to MES Acquisition Function Constructors
Browse files Browse the repository at this point in the history
Summary: Different acquisition functions take different kwargs as inputs into their constructors. To standardize the inputs, we add `**kwargs` to the constructors, specifically for `qMaxValueEntropy` and `qMultiFidelityMaxValueEntropy`.

Reviewed By: Balandat, qingfeng10

Differential Revision: D22460703

fbshipit-source-id: dc750945d23aefd8c59edd7528b497ef0df421c2
  • Loading branch information
EricZLou authored and facebook-github-bot committed Jul 13, 2020
1 parent fd24aee commit e5b6935
Showing 1 changed file with 3 additions and 1 deletion.
4 changes: 3 additions & 1 deletion botorch/acquisition/max_value_entropy_search.py
Original file line number Diff line number Diff line change
Expand Up @@ -25,7 +25,7 @@

from copy import deepcopy
from math import log
from typing import Callable, Optional
from typing import Any, Callable, Optional

import torch
from botorch.acquisition.cost_aware import CostAwareUtility, InverseCostWeightedUtility
Expand Down Expand Up @@ -70,6 +70,7 @@ def __init__(
use_gumbel: bool = True,
maximize: bool = True,
X_pending: Optional[Tensor] = None,
**kwargs: Any,
) -> None:
r"""Single-outcome max-value entropy search acquisition function.
Expand Down Expand Up @@ -336,6 +337,7 @@ def __init__(
cost_aware_utility: Optional[CostAwareUtility] = None,
project: Callable[[Tensor], Tensor] = lambda X: X,
expand: Callable[[Tensor], Tensor] = lambda X: X,
**kwargs: Any,
) -> None:
r"""Single-outcome max-value entropy search acquisition function.
Expand Down

0 comments on commit e5b6935

Please sign in to comment.