Skip to content

Commit b5ae882

Browse files
committed
add type hints for base acquisition get/set functions
1 parent cf87c7b commit b5ae882

File tree

1 file changed

+2
-2
lines changed

1 file changed

+2
-2
lines changed

bayes_opt/acquisition.py

Lines changed: 2 additions & 2 deletions
Original file line numberDiff line numberDiff line change
@@ -109,7 +109,7 @@ def _fit_gp(self, gp: GaussianProcessRegressor, target_space: TargetSpace) -> No
109109
if target_space.constraint is not None:
110110
target_space.constraint.fit(target_space.params, target_space._constraint_values)
111111

112-
def get_acquisition_params(self):
112+
def get_acquisition_params(self) -> dict[str, Any] | NoReturn:
113113
"""
114114
Get the parameters of the acquisition function.
115115
@@ -123,7 +123,7 @@ def get_acquisition_params(self):
123123
)
124124
raise NotImplementedError(error_msg)
125125

126-
def set_acquisition_params(self, **params):
126+
def set_acquisition_params(self, **params) -> None | NoReturn:
127127
"""
128128
Set the parameters of the acquisition function.
129129

0 commit comments

Comments
 (0)