Fix mypy error

This commit is contained in:
Contramundum committed 2023-08-31 12:24:01 +09:00
1 parent ee02294381
commit 2e05d2c53d
1 file changed
+1 -1
+1 -1
View File
@@ -222,7 +222,7 @@ class _PreferentialGP:
last_params = [p.detach().clone() for p in optim.param_groups[0]["params"]]
for _ in range(max_iter):
def closure():
def closure() -> Tensor:
optim.zero_grad()
noise = self.log_noise.exp()
cov0 = self.kernel.forward(X, X).to_dense()