diff --git a/pyproject.toml b/pyproject.toml
index f4460c8..ccd8ae4 100644
--- a/pyproject.toml
+++ b/pyproject.toml
@@ -39,7 +39,7 @@ train = [
   "torch-scatter",
   "torch-sparse",
   "torch-cluster",
-  "pytorch-lightning==2.4.0",
+  "pytorch-lightning==2.5.0.post0",
   "jsonargparse[signatures]"
 ]