mirrors/rocm_jax
1
0
Fork 0
mirror of https://github.com/ROCm/jax.git synced 2025-04-16 03:46:06 +00:00
Code Issues Packages Projects Releases Wiki Activity

CI: temporarily pin numpy to <1.21

Browse Source
This commit is contained in:
Jake VanderPlas 2021-06-22 11:15:16 -07:00
parent 6382cee708
commit 0c91be7b46
2 changed files with 4 additions and 1 deletions
Show Stats Download Patch File Download Diff File Expand all files Collapse all files

3
build/test-requirements.txt
View File

@ -1,7 +1,8 @@
flake8
flatbuffers==1.12
# For now, we pin the numpy version here
numpy>=1.17
# TODO(jakevdp): unpin maximum version when minimum jaxlib supports newer numpy
numpy>=1.17,<1.21
mypy==0.902
pillow
pytest-benchmark

2
docs/requirements.txt
View File

@ -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.