Reduce test flakiness.

PiperOrigin-RevId: 527637662
google · Apr 27, 2023 · 40042af · 40042af
1 parent 736806d
commit 40042af
Showing 1 changed file with 2 additions and 2 deletions.
diff --git a/vizier/utils/profiler_test.py b/vizier/utils/profiler_test.py
@@ -103,8 +103,8 @@ def selu(x, alpha=1.67, lambda_=1.05):
     selu(x)
     self.assertLen(profiler.Storage().runtimes().get('selu'), 2)
     jt_microseconds = profiler.Storage().runtimes().get('selu')[1].microseconds
-    # Should be at least 10 times faster.
-    self.assertLess(jt_microseconds, microseconds / 10)
+    # Should be at least 7 times faster (rough estimate so it's not flaky).
+    self.assertLess(jt_microseconds, microseconds / 7)
 
   def test_with_jax_reverse_decorator_order(self):
     # If jax.jit decorator is first, only the first runtime is recorded.