CI: temporarily pin numpy to <1.21

jax-ml · Jun 22, 2021 · ae407c6 · ae407c6
1 parent 6382cee
commit ae407c6
Show file tree

Hide file tree

Showing 2 changed files with 3 additions and 1 deletion.
diff --git a/build/test-requirements.txt b/build/test-requirements.txt
@@ -1,7 +1,7 @@
 flake8
 flatbuffers==1.12
 # For now, we pin the numpy version here
-numpy>=1.17
+numpy>=1.17,<1.21  # TODO(jakevdp): unpin maximum version when jaxlib is fixed
 mypy==0.902
 pillow
 pytest-benchmark

diff --git a/docs/requirements.txt b/docs/requirements.txt
@@ -13,4 +13,6 @@ pytest-xdist
 # Packages used for notebook execution
 matplotlib
 scikit-learn
+# TODO(jakevdp) remove numpy pinning when minimum jaxlib supports newer numpy.
+numpy<1.21
 .[cpu]  # Install jax from the current directory; jaxlib from pypi.