diff --git a/tests/constraints.txt b/tests/constraints.txt index 7abffee14c8ce2..065aa73a0d636b 100644 --- a/tests/constraints.txt +++ b/tests/constraints.txt @@ -21,6 +21,6 @@ pytest>=5.0,<7.5 pytest-dependency==0.5.1 pytest-html==3.2.0 pytest-timeout==2.1.0 -jax<=0.4.14 +jax<=0.4.19 jaxlib<=0.4.14 torch<2.1.0 \ No newline at end of file