diff --git a/tests/constraints.txt b/tests/constraints.txt
index 7a32f3987fa5bb..83a1ad6e4d1e27 100644
--- a/tests/constraints.txt
+++ b/tests/constraints.txt
@@ -24,4 +24,4 @@ pytest-html==4.1.1
 pytest-timeout==2.1.0
 jax<=0.4.14
 jaxlib<=0.4.14
-torch>=1.13,<2.2
+torch>=1.13,<2.6