Fix test_optimize_l0_homotopy test numerics (facebook#2493)

Summary: Before this change, this test fails (reproducibly) on my M1 Mac. Pull Request resolved: facebook#2493 Reviewed By: dme65 Differential Revision: D58068980 Pulled By: Balandat fbshipit-source-id: cda35c9117d63bd31e523652c4db7674f9b22cb0
bernardbeckerman · Jun 3, 2024 · 1f94ac8 · 1f94ac8
1 parent c410658
commit 1f94ac8
Showing 1 changed file with 2 additions and 2 deletions.
diff --git a/ax/models/torch/tests/test_sebo.py b/ax/models/torch/tests/test_sebo.py
@@ -254,8 +254,8 @@ def test_optimize_l0_homotopy(
                 "raw_samples": 16,
             },
         )
-        self.assertEqual(candidate, torch.zeros(1, **tkwargs))
-        self.assertEqual(acqf_val, 5 * torch.ones(1, **tkwargs))
+        self.assertTrue(torch.allclose(candidate, torch.zeros(1, **tkwargs)))
+        self.assertTrue(torch.allclose(acqf_val, 5 * torch.ones(1, **tkwargs)))
         self.assertEqual(weights, torch.ones(1, **tkwargs))
 
     @mock.patch(f"{SEBOACQUISITION_PATH}.optimize_acqf_homotopy")