diff --git a/website/pages/en/index.js b/website/pages/en/index.js index 3b8661d78a..1e17ec0520 100755 --- a/website/pages/en/index.js +++ b/website/pages/en/index.js @@ -118,7 +118,7 @@ from botorch.fit import fit_gpytorch_mll from botorch.utils import standardize from gpytorch.mlls import ExactMarginalLogLikelihood -train_X = torch.rand(10, 2) +train_X = torch.rand(10, 2, dtype=torch.double) Y = 1 - torch.linalg.norm(train_X - 0.5, dim=-1, keepdim=True) Y = Y + 0.1 * torch.randn_like(Y) # add some noise train_Y = standardize(Y)