Fix lint error at head.

skondla · Jul 3, 2024 · f408d2a · f408d2a
1 parent dffd72e
commit f408d2a
Showing 1 changed file with 0 additions and 1 deletion.
diff --git a/jax/_src/ad_checkpoint.py b/jax/_src/ad_checkpoint.py
@@ -32,7 +32,6 @@
 from jax._src import effects
 from jax._src import source_info_util
 from jax._src import traceback_util
-from jax._src import util
 from jax._src.api_util import flatten_fun, shaped_abstractify
 from jax._src.interpreters import ad
 from jax._src.interpreters import batching