Deflake the benchmarks tests.

PiperOrigin-RevId: 612560615

vizier/_src/benchmarks/analyzers/convergence_curve_test.py

@@ -294,7 +294,9 @@ def test_convert_with_origin_reference(self):
 
  curve = generator.convert(pytrials)
  np.testing.assert_array_equal(curve.xs, [1, 2, 3])
- np.testing.assert_array_almost_equal(curve.ys, [[0.0, 3.0, 8.0]], decimal=1)
+ np.testing.assert_array_almost_equal(
+ curve.ys, [[0.0, 3.0, 8.0]], decimal=0.5
+ )
 
  def test_convert_with_reference(self):
  generator = convergence.HypervolumeCurveConverter(
@@ -327,7 +329,9 @@ def test_convert_with_reference(self):
 
  curve = generator.convert(pytrials)
  np.testing.assert_array_equal(curve.xs, [1, 2, 3])
- np.testing.assert_array_almost_equal(curve.ys, [[0.0, 0.0, 2.0]], decimal=1)
+ np.testing.assert_array_almost_equal(
+ curve.ys, [[0.0, 0.0, 2.0]], decimal=0.5
+ )
 
  def test_convert_with_none_reference(self):
  generator = convergence.HypervolumeCurveConverter([
@@ -357,7 +361,9 @@ def test_convert_with_none_reference(self):
 
  curve = generator.convert(pytrials)
  np.testing.assert_array_equal(curve.xs, [1, 2, 3])
- np.testing.assert_array_almost_equal(curve.ys, [[0.0, 0.0, 1.0]], decimal=1)
+ np.testing.assert_array_almost_equal(
+ curve.ys, [[0.0, 0.0, 1.0]], decimal=0.5
+ )
 
  def test_convert_with_inf_none_reference(self):
  generator = convergence.HypervolumeCurveConverter([
@@ -433,7 +439,7 @@ def test_convert_with_state(self):
 
  curve = generator.convert(pytrials)
  np.testing.assert_array_equal(curve.xs, [4, 5])
- np.testing.assert_array_almost_equal(curve.ys, [[9.0, 10.0]], decimal=1)
+ np.testing.assert_array_almost_equal(curve.ys, [[9.0, 10.0]], decimal=0.5)
 
  def test_convert_factor_with_inf(self):
  generator = convergence.HypervolumeCurveConverter(
@@ -466,7 +472,9 @@ def test_convert_factor_with_inf(self):
 
  curve = generator.convert(pytrials)
  np.testing.assert_array_equal(curve.xs, [1, 2, 3])
- np.testing.assert_array_almost_equal(curve.ys, [[0.0, 1.0, 4.0]], decimal=1)
+ np.testing.assert_array_almost_equal(
+ curve.ys, [[0.0, 1.0, 4.0]], decimal=0.5
+ )
 
 
 class MultiMetricCurveConverterTest(parameterized.TestCase):
@@ -544,7 +552,9 @@ def test_convert_multiobjective(self):
 
  curve = generator.convert(pytrials)
  np.testing.assert_array_equal(curve.xs, [1, 2, 3])
- np.testing.assert_array_almost_equal(curve.ys, [[4.0, 4.0, 8.0]], decimal=1)
+ np.testing.assert_array_almost_equal(
+ curve.ys, [[4.0, 4.0, 8.0]], decimal=0.5
+ )
 
 
 class RestartingCurveConverterTest(absltest.TestCase):
@@ -584,7 +594,9 @@ def converter_factory():
 
  curve = restart_converter.convert(pytrials)
  np.testing.assert_array_equal(curve.xs, [1, 2, 3])
- np.testing.assert_array_almost_equal(curve.ys, [[0.0, 5.0, 9.0]], decimal=1)
+ np.testing.assert_array_almost_equal(
+ curve.ys, [[0.0, 5.0, 9.0]], decimal=0.5
+ )
 
  pytrials = []
  pytrials.append(
@@ -638,7 +650,9 @@ def converter_factory():
 
  curve = converter.convert(pytrials)
  np.testing.assert_array_equal(curve.xs, [1, 2, 3])
- np.testing.assert_array_almost_equal(curve.ys, [[0.0, 5.0, 9.0]], decimal=1)
+ np.testing.assert_array_almost_equal(
+ curve.ys, [[0.0, 5.0, 9.0]], decimal=0.5
+ )
 
  pytrials = []
  pytrials.append(