diff --git a/botorch/test_functions/multi_fidelity.py b/botorch/test_functions/multi_fidelity.py index 703e30ca3b..22c593f292 100644 --- a/botorch/test_functions/multi_fidelity.py +++ b/botorch/test_functions/multi_fidelity.py @@ -84,6 +84,7 @@ def __init__( Args: noise_std: Standard deviation of the observation noise. negate: If True, negate the function. + dtype: The dtype that is used for the bounds of the function. """ super().__init__(noise_std=noise_std, negate=negate, dtype=dtype) self.register_buffer("ALPHA", torch.tensor([1.0, 1.2, 3.0, 3.2])) @@ -142,6 +143,7 @@ def __init__( dim: The (input) dimension. Must be at least 3. noise_std: Standard deviation of the observation noise. negate: If True, negate the function. + dtype: The dtype that is used for the bounds of the function. """ if dim < 3: raise ValueError( diff --git a/botorch/test_functions/synthetic.py b/botorch/test_functions/synthetic.py index 91c1df843e..b4efc0920c 100644 --- a/botorch/test_functions/synthetic.py +++ b/botorch/test_functions/synthetic.py @@ -271,6 +271,7 @@ def __init__( dim: The (input) dimension. noise_std: Standard deviation of the observation noise. negate: If True, negate the function. + dtype: The dtype that is used for the bounds of the function. """ self.dim = dim if bounds is None: