From 89d25bb1a3a90367eeca6ae8d942ddf6b7a07ff4 Mon Sep 17 00:00:00 2001 From: Peter Hawkins Date: Thu, 9 May 2024 07:08:48 -0700 Subject: [PATCH] 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 --- examples/examples_test.py | 3 +++ 1 file changed, 3 insertions(+) diff --git a/examples/examples_test.py b/examples/examples_test.py index c9cb2991c030..fd705a4ef799 100644 --- 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