diff --git a/tests/e2e/mnist_pip_requirements.txt b/tests/e2e/mnist_pip_requirements.txt
index 60811f18a..07123da4d 100644
--- a/tests/e2e/mnist_pip_requirements.txt
+++ b/tests/e2e/mnist_pip_requirements.txt
@@ -1,4 +1,4 @@
-pytorch_lightning==1.9.5
+pytorch_lightning==2.5.0.post0
 torchmetrics==0.9.1
 torchvision==0.20.1
 minio