a

EnzymeAD · Oct 16, 2024 · 0da2ac0 · 0da2ac0
1 parent b529739
commit 0da2ac0
Show file tree

Hide file tree

Showing 2 changed files with 3 additions and 1 deletion.
diff --git a/builddeps/requirements.in b/builddeps/requirements.in
@@ -6,3 +6,4 @@
 jax >= 0.4.21
 jaxlib >= 0.4.21
 absl_py >= 2.0.0
+ortools >= 9.10
diff --git a/test/BUILD b/test/BUILD
@@ -61,10 +61,11 @@ TEST_DEPS = [
         "//:enzyme_ad",
         "@pypi_jax//:pkg",
         "@pypi_absl_py//:pkg",
+        "@pypi_ortools//:pkg",
     ] + select({
         ":use_tpu": ["@pypi_libtpu_nightly//:pkg", "@pypi_requests//:pkg"],
         "@bazel_tools//src/conditions:linux_x86_64": ["@pypi_jax_cuda12_plugin//:pkg", "@pypi_nvidia_cuda_nvrtc_cu12//:pkg"],
-        "//conditions:default": ["@pypi_ortools//:pkg"]
+        "//conditions:default": [],
     })
 
 [