diff --git a/tests/constraints.txt b/tests/constraints.txt
index 97c88475f17..671c60bc937 100644
--- a/tests/constraints.txt
+++ b/tests/constraints.txt
@@ -22,3 +22,4 @@ pytest-html==3.2.0
 pytest-timeout==2.1.0
 jax<=0.4.14
 jaxlib<=0.4.14
+torch<2.1.0
\ No newline at end of file