Update optuna/_gp/acqf.py

Co-authored-by: Shuhei Watanabe <47781922+nabenabe0928@users.noreply.github.com>
Alnusjaponica · Feb 8, 2024 · dd030cd · dd030cd
1 parent 841e7cb
commit dd030cd
Showing 1 changed file with 1 addition and 1 deletion.
diff --git 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."