use absltest for tests

JAX already depends on absltest so we have it "for free" as a dependency. Using a test framework composes better with build infrastructure.
EnzymeAD · Dec 15, 2023 · 5378879 · 5378879
1 parent 9c89a54
commit 5378879
Show file tree

Hide file tree

Showing 4 changed files with 361 additions and 344 deletions.
diff --git a/BUILD b/BUILD
@@ -42,6 +42,7 @@ py_wheel(
     deps = ["//src/enzyme_ad/jax:enzyme_jax_internal", ":enzyme_jax_data"],
     strip_path_prefixes = ["src/"],
     requires = [
+        "absl_py >= 2.0.0",
         "jax >= 0.4.21",
         "jaxlib >= 0.4.21",
     ],