diff --git a/tests/constraints.txt b/tests/constraints.txt index f09da0d3b409e9..ca4553a0be2b1a 100644 --- a/tests/constraints.txt +++ b/tests/constraints.txt @@ -23,7 +23,7 @@ pytest-dependency==0.5.1 pytest-html==4.1.1 pytest-timeout==2.3.1 jax<=0.4.33 -jaxlib<=0.4.33 +jaxlib<=0.4.34 kornia==0.7.0 networkx<=3.3 flax<=0.9.0