From 6ddd51f5cd9be258a2d27709d2c10e60781e9fe7 Mon Sep 17 00:00:00 2001 From: "Alexander V. Hopp" Date: Tue, 29 Oct 2024 15:40:58 +0100 Subject: [PATCH] Add missing docstring args --- botorch/test_functions/multi_fidelity.py | 2 ++ botorch/test_functions/synthetic.py | 1 + 2 files changed, 3 insertions(+) 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: