From 348d92e0125b11b235b775e8c0f35ff0cd57eae3 Mon Sep 17 00:00:00 2001 From: contramundum53 Date: Fri, 8 Sep 2023 14:10:47 +0900 Subject: [PATCH] Update optuna_dashboard/preferential/samplers/gp.py Co-authored-by: Naoto Mizuno --- optuna_dashboard/preferential/samplers/gp.py | 2 +- 1 file changed, 1 insertion(+), 1 deletion(-) diff --git a/optuna_dashboard/preferential/samplers/gp.py b/optuna_dashboard/preferential/samplers/gp.py index 4fafe98a..2b7881c7 100644 --- a/optuna_dashboard/preferential/samplers/gp.py +++ b/optuna_dashboard/preferential/samplers/gp.py @@ -194,7 +194,7 @@ def _orthants_MVN_EP( mu = mu - Sxy * ((db + mean1 * da) * dr) cov = cov - (Sxy[:, None] * (da * dr)) @ Sxy[None, :] log_zs[i] = logz - return (mu, cov, torch.sum(log_zs)) + return mu, cov, torch.sum(log_zs) _orthants_MVN_EP_jit = torch.jit.script(_orthants_MVN_EP)