diff --git a/tests/constraints.txt b/tests/constraints.txt
index f3569f126cc3fb..cc1b4328c7e5ab 100644
--- a/tests/constraints.txt
+++ b/tests/constraints.txt
@@ -26,4 +26,4 @@ pytest-html==4.1.1
 pytest-timeout==2.2.0
 jax<=0.4.14
 jaxlib<=0.4.14
-torch>=1.13,<2.3
+torch>=1.13,<2.6