From 0153b52b0b19cddf08ba55b96b7bc8ed9aa916ab Mon Sep 17 00:00:00 2001 From: Giacomo Petrillo Date: Tue, 28 Mar 2023 01:42:00 +0200 Subject: [PATCH] maint: fix jax version specifier --- setup.py | 4 ++-- 1 file changed, 2 insertions(+), 2 deletions(-) diff --git a/setup.py b/setup.py index 11d7880c..79199809 100644 --- a/setup.py +++ b/setup.py @@ -50,8 +50,8 @@ install_requires=[ 'numpy>=1.20', # first version with broadcast_shapes 'scipy>=1.5', # minimum required by jax - 'jax>=0.4.1<=0.4.6', # 0.4.1 first supporting type dispatching on jax arrays - 'jaxlib>=0.4.1<=0.4.6', # 0.4.6 last supporting old jit implementation, the new one triggers a leaked tracer exception in linalg operations + 'jax>=0.4.1,<=0.4.6', # 0.4.1 first supporting type dispatching on jax arrays + 'jaxlib>=0.4.1,<=0.4.6', # 0.4.6 last supporting old jit implementation, the new one triggers a leaked tracer exception in linalg operations 'gvar>=11.10.1', # first supporting gvar.gvar(x, cov) with readonly cov ] )