From dd030cd227b41b49d263fe1611e1e0486c9b3e6e Mon Sep 17 00:00:00 2001 From: contramundum53 Date: Thu, 8 Feb 2024 13:23:37 +0900 Subject: [PATCH] Update optuna/_gp/acqf.py Co-authored-by: Shuhei Watanabe <47781922+nabenabe0928@users.noreply.github.com> --- optuna/_gp/acqf.py | 2 +- 1 file changed, 1 insertion(+), 1 deletion(-) diff --git a/optuna/_gp/acqf.py b/optuna/_gp/acqf.py index b43209e02e..82fa62873b 100644 --- a/optuna/_gp/acqf.py +++ b/optuna/_gp/acqf.py @@ -120,7 +120,7 @@ def eval_acqf(acqf_params: AcquisitionFunctionParams, x: torch.Tensor) -> torch. if acqf_params.acqf_type == AcquisitionFunctionType.LOG_EI: return logei(mean=mean, var=var + acqf_params.acqf_stabilizing_noise, f0=acqf_params.max_Y) elif acqf_params.acqf_type == AcquisitionFunctionType.UCB: - assert acqf_params.beta is not None + assert acqf_params.beta is not None, "beta must be given to UCB." return ucb(mean=mean, var=var, beta=acqf_params.beta) else: assert False, "Unknown acquisition function type."