Reenable examples_test in Bazel build.

Fix bitrot. This test was disabled years ago because it was slow, but it isn't any more. PiperOrigin-RevId: 632138101
jax-ml · May 9, 2024 · 89d25bb · 89d25bb
1 parent 1e88e2f
commit 89d25bb
Showing 1 changed file with 3 additions and 0 deletions.
diff --git a/examples/examples_test.py b/examples/examples_test.py
@@ -26,6 +26,9 @@
 from jax import lax
 from jax import random
 import jax.numpy as jnp
+from jax._src import test_util as jtu
+
+del jtu  # Needed for flags
 
 sys.path.append(os.path.dirname(os.path.dirname(os.path.abspath(__file__))))
 from examples import kernel_lsq