Update test/optim/test_optimize.py

Co-authored-by: Max Balandat <Balandat@users.noreply.github.com>
pytorch · Jul 3, 2023 · a4445be · a4445be
1 parent 3cf28b1
commit a4445be
Showing 1 changed file with 1 addition and 1 deletion.
diff --git a/test/optim/test_optimize.py b/test/optim/test_optimize.py
@@ -875,7 +875,7 @@ def nlc4(x):
             self.assertTrue(
                 torch.allclose(
                     candidates,
-                    torch.tensor([[2, 1, 1]], **tkwargs),
+                    torch.tensor([[2.0, 1.0, 1.0]], **tkwargs),
                 )
             )
             self.assertTrue(