Skip to content

Commit 079a758

Browse files
committed
linting
1 parent c1dd52e commit 079a758

File tree

1 file changed

+3
-3
lines changed

1 file changed

+3
-3
lines changed

lume_model/models/gp_model.py

Lines changed: 3 additions & 3 deletions
Original file line numberDiff line numberDiff line change
@@ -337,9 +337,9 @@ def _check_covariance_matrix(self, cov: torch.Tensor) -> torch.Tensor:
337337
torch.linalg.cholesky(cov)
338338
except torch._C._LinAlgError:
339339
warnings.warn(
340-
f"Covariance matrix is not positive definite. Attempting to add jitter the diagonal."
340+
"Covariance matrix is not positive definite. Attempting to add jitter the diagonal."
341341
)
342-
l = psd_safe_cholesky(cov) # determines jitter iteratively
343-
cov = l @ l.transpose(-1, -2)
342+
lm = psd_safe_cholesky(cov) # determines jitter iteratively
343+
cov = lm @ lm.transpose(-1, -2)
344344

345345
return cov

0 commit comments

Comments
 (0)