diff --git a/goldenretriever/version.py b/goldenretriever/version.py
index af0b8fb..3a517ca 100644
--- a/goldenretriever/version.py
+++ b/goldenretriever/version.py
@@ -4,7 +4,7 @@
 _MINOR = "9"
 # On main and in a nightly release the patch should be one ahead of the last
 # released build.
-_PATCH = "3"
+_PATCH = "4"
 # This is mainly for nightly builds which have the suffix ".dev$DATE". See
 # https://semver.org/#is-v123-a-semantic-version for the semantics.
 _SUFFIX = os.environ.get("GOLDENRETRIEVER_VERSION_SUFFIX", "")
diff --git a/requirements.txt b/requirements.txt
index e636c45..4432412 100644
--- a/requirements.txt
+++ b/requirements.txt
@@ -7,7 +7,7 @@ rich>=13.0.0,<14.0.0
 scikit-learn>=1.3,<1.5
 
 # train
-lightning>=2.1,<2.2
+lightning>=2.1,<2.3
 hydra-core>=1.3,<1.4
 hydra_colorlog
 wandb>=0.16,<0.17