Bump lightning from 2.2.5 to 2.4.0 (#102)

* Bump lightning from 2.2.5 to 2.4.0

Bumps [lightning](https://github.com/Lightning-AI/lightning) from 2.2.5 to 2.4.0.
- [Release notes](https://github.com/Lightning-AI/lightning/releases)
- [Commits](Lightning-AI/pytorch-lightning@2.2.5...2.4.0)

---
updated-dependencies:
- dependency-name: lightning
  dependency-type: direct:production
  update-type: version-update:semver-minor
...

Signed-off-by: dependabot[bot] <[email protected]>

* be more flexible with lightning reqs

Signed-off-by: Carlos Gomes <[email protected]>

---------

Signed-off-by: dependabot[bot] <[email protected]>
Signed-off-by: Carlos Gomes <[email protected]>
Co-authored-by: dependabot[bot] <49699333+dependabot[bot]@users.noreply.github.com>
Co-authored-by: Carlos Gomes <[email protected]>

pyproject.toml

@@ -44,7 +44,7 @@ dependencies = [
   "geobench>=1.0.0",
   "mlflow>=2.12.1",
   # broken due to https://github.com/Lightning-AI/pytorch-lightning/issues/19977
-  "lightning>=2, <=2.2.5",
+  "lightning[pytorch-extra]>=2,!=2.3.*",
 ]
 
 [project.optional-dependencies]

requirements/required.txt

@@ -10,7 +10,7 @@ lightly==1.4.25
 h5py==3.10.0
 geobench==1.0.0
 mlflow==2.14.3
-lightning==2.2.5
+lightning==2.4.0
 
 # These dependencies are optional
 # and must be installed just in case 
@@ -23,4 +23,4 @@ lightning==2.2.5
 #openmim
 
 # Lastly, install mmsegmentation using mim:
-# mim mmsegmentation
+# mim mmsegmentation