Relaxed test tolerance

PiperOrigin-RevId: 725193038
google · Feb 10, 2025 · e119b7b · e119b7b
1 parent 7ce0ae8
commit e119b7b
Showing 1 changed file with 2 additions and 1 deletion.
diff --git a/vizier/_src/algorithms/designers/gp_ucb_pe_test.py b/vizier/_src/algorithms/designers/gp_ucb_pe_test.py
@@ -328,7 +328,8 @@ def test_on_flat_space(
           # single-metric case because the acquisition value in the multi-metric
           # case is randomly scalarized.
           if num_metrics == 1:
-            self.assertAlmostEqual(mean + 10.0 * stddev_from_all, acq)
+            expected_acq = mean + 10.0 * stddev_from_all
+            self.assertAlmostEqual(expected_acq, acq, delta=1e-6)
           self.assertTrue(use_ucb)
           continue