diff --git a/tests/constraints.txt b/tests/constraints.txt index 4e3455dd1b8b27..b6d18b39b59d5f 100644 --- a/tests/constraints.txt +++ b/tests/constraints.txt @@ -23,5 +23,5 @@ pytest-dependency==0.5.1 pytest-html==4.1.1 pytest-timeout==2.1.0 jax<=0.4.14 -jaxlib<=0.4.14 +jaxlib<=0.4.23 torch>=1.13,<2.2