diff --git a/python/requirements/ml/tune-test-requirements.txt b/python/requirements/ml/tune-test-requirements.txt
index 3a929149b65e..d00a1ce7fccc 100644
--- a/python/requirements/ml/tune-test-requirements.txt
+++ b/python/requirements/ml/tune-test-requirements.txt
@@ -6,7 +6,7 @@ jupyterlab==3.6.1
 matplotlib!=3.4.3
 
 pytest-remotedata==0.3.2
-pytorch-lightning==1.8.6
+pytorch-lightning==2.6.1
 fairscale==0.4.6
 shortuuid==1.0.1
 timm==0.9.2