diff --git a/tests/constraints.txt b/tests/constraints.txt
index 5ecacb4b65c03e..1e39cc389487be 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.5