From e08f2e8e03d259d9f1f222a9c1b2dff465c6a5f2 Mon Sep 17 00:00:00 2001 From: Jose Pablo Folch Date: Tue, 10 Sep 2024 22:44:09 +0100 Subject: [PATCH] fixed linting --- bofire/strategies/predictives/botorch.py | 2 +- 1 file changed, 1 insertion(+), 1 deletion(-) diff --git a/bofire/strategies/predictives/botorch.py b/bofire/strategies/predictives/botorch.py index 930b47c7..e9863d5a 100644 --- a/bofire/strategies/predictives/botorch.py +++ b/bofire/strategies/predictives/botorch.py @@ -173,7 +173,7 @@ def _predict(self, transformed: pd.DataFrame) -> Tuple[np.ndarray, np.ndarray]: # observation noise is not implemented for MultiTaskGPSurrogate, has to be treated differently if self.surrogate_specs.surrogates[0].type == "MultiTaskGPSurrogate": posterior = self.model.posterior(X=X, observation_noise=False) # type: ignore - likelihood_noise = self.model.likelihood.noise.cpu().detach().numpy() + likelihood_noise = self.model.likelihood.noise.cpu().detach().numpy() # type: ignore else: posterior = self.model.posterior(X=X, observation_noise=True) # type: ignore likelihood_noise = 0