From 868c27da3697460a345d920c37e99e6cca111dcf Mon Sep 17 00:00:00 2001 From: Matthew Feickert Date: Fri, 26 Aug 2022 21:40:53 -0500 Subject: [PATCH] build: Update lower bounds to tensorflow v2.7.0, jaxlib v0.1.61 (#1962) * Update lower bound of the supported tensorflow versions to v2.7.0 to ensure numpy v1.20.0+ is available. - numpy.typing is needed for PR #1940 and was added in numpy v1.20.0. - tensorflow v2.6.5 restricts numpy to 'numpy~=1.19.2' and tensorflow v2.7.0 relaxes numpy constraints to 'numpy>=1.14.5'. * Update lower bound of the supported jaxlib versions to v0.1.61. - jaxlib v0.1.60 required 'numpy>=1.12,<1.20' and jaxlib v0.1.61 relaxes this to 'numpy>=1.16'. * Update tests/constraints.txt to use tensorflow==2.7.0 and jaxlib==0.1.61. --- setup.py | 4 ++-- tests/constraints.txt | 4 ++-- 2 files changed, 4 insertions(+), 4 deletions(-) diff --git a/setup.py b/setup.py index 401f2ea8ae..1406f9b4ff 100644 --- a/setup.py +++ b/setup.py @@ -3,11 +3,11 @@ extras_require = { 'shellcomplete': ['click_completion'], 'tensorflow': [ - 'tensorflow>=2.6.5', # c.f. PR #1874 + 'tensorflow>=2.7.0', # c.f. PR #1962 'tensorflow-probability>=0.11.0', # c.f. PR #1657 ], 'torch': ['torch>=1.10.0'], # c.f. PR #1657 - 'jax': ['jax>=0.2.10', 'jaxlib>=0.1.60,!=0.1.68'], # c.f. Issue 1501 + 'jax': ['jax>=0.2.10', 'jaxlib>=0.1.61,!=0.1.68'], # c.f. PR #1962, Issue #1501 'xmlio': ['uproot>=4.1.1'], # c.f. PR #1567 'minuit': ['iminuit>=2.7.0'], # c.f. PR #1895 } diff --git a/tests/constraints.txt b/tests/constraints.txt index 054dd7b32d..3e21e026a7 100644 --- a/tests/constraints.txt +++ b/tests/constraints.txt @@ -12,7 +12,7 @@ uproot==4.1.1 # minuit iminuit==2.7.0 # c.f. PR #1895 # tensorflow -tensorflow==2.6.5 # c.f. PR #1874 +tensorflow==2.7.0 # c.f. PR #1962 tensorflow-probability==0.11.0 # c.f. PR #1657 # torch torch==1.10.0 @@ -21,4 +21,4 @@ torch==1.10.0 # c.f. https://github.com/google/jax/discussions/7608#discussioncomment-1269342 --find-links https://storage.googleapis.com/jax-releases/jax_releases.html jax==0.2.10 -jaxlib==0.1.60 # jax v0.2.10 requires jaxlib>=0.1.60 +jaxlib==0.1.61 # c.f. PR #1962