diff --git a/tests/constraints.txt b/tests/constraints.txt
index b964d71d91509f..59b17655079fc0 100644
--- a/tests/constraints.txt
+++ b/tests/constraints.txt
@@ -26,7 +26,7 @@ jax<=0.4.14
 jaxlib<=0.4.14
 kornia==0.7.0
 networkx<=3.1.0
-keras>=2.0.0,<3.0.0
+keras>=2.0.0,<4.0.0
 
 --extra-index-url https://download.pytorch.org/whl/cpu
 torch>=1.13,<2.3
\ No newline at end of file