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