Update test-requirements.txt

EnzymeAD · Nov 6, 2024 · 58a6d19 · 58a6d19
1 parent 140da3b
commit 58a6d19
Showing 1 changed file with 1 addition and 1 deletion.
diff --git a/builddeps/test-requirements.txt b/builddeps/test-requirements.txt
@@ -1,11 +1,11 @@
 absl-py
-jax
 numpy
 jaxlib
 https://github.com/wsmoses/jax-md/archive/45059b8f63dad0b5cb171feafff71b82162487e7.tar.gz
 # maxtext can't be installed concurrently, but installing it fixes
 # https://github.com/wsmoses/maxtext/archive/bc50722be7d89e4003bd830b80e4ac968be658eb.tar.gz
 jax[cuda12_pip]; sys_platform == 'linux'
+jax; sys_platform != 'linux'
 requests; sys_platform == 'linux'
 # -f https://storage.googleapis.com/jax-releases/libtpu_releases.html
 # libtpu-nightly == 0.1.dev20240729; sys_platform == 'linux'