diff --git a/pyproject.toml b/pyproject.toml
index 3002358..f2f00a3 100644
--- a/pyproject.toml
+++ b/pyproject.toml
@@ -60,7 +60,7 @@ dependencies = [
     "moviepy ~=1.0.3",
     "rlax ~=0.1.5",
     "farconf @ git+https://github.com/AlignmentResearch/farconf.git",
-    "ray[tune] ~=2.40.0",
+    "ray[tune] >=2.40,<2.57",
     "matplotlib ~=3.9.0",
     "jax ==0.4.34",
     "jaxlib ==0.4.34",
diff --git a/requirements.txt b/requirements.txt
index 6b32ae7..b41b275 100644
--- a/requirements.txt
+++ b/requirements.txt
@@ -284,7 +284,7 @@ pyyaml==6.0.2
     #   pre-commit
     #   ray
     #   wandb
-ray[tune]==2.40.0
+ray[tune]==2.56.0
     # via train-learned-planner (pyproject.toml)
 referencing==0.35.1
     # via