Add missing docstring args

pytorch · Oct 29, 2024 · 6ddd51f · 6ddd51f
1 parent b9f1cf5
commit 6ddd51f
Show file tree

Hide file tree

Showing 2 changed files with 3 additions and 0 deletions.
diff --git 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
@@ -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: