From c136c8b0191ecf2c51395adb9697f7045e82d916 Mon Sep 17 00:00:00 2001
From: Lorenzo Stella <stellalo@amazon.com>
Date: Fri, 14 Jun 2024 15:16:44 +0200
Subject: [PATCH] bump pytorch lightning compat

---
 requirements/requirements-pytorch.txt | 4 ++--
 1 file changed, 2 insertions(+), 2 deletions(-)

diff --git a/requirements/requirements-pytorch.txt b/requirements/requirements-pytorch.txt
index 683bb101d7..e3e9432c0d 100644
--- a/requirements/requirements-pytorch.txt
+++ b/requirements/requirements-pytorch.txt
@@ -1,6 +1,6 @@
 torch>=1.9,<3
-lightning>=2.0,<2.2
+lightning>=2.2.2,<2.4
 # Capping `lightning` does not cap `pytorch_lightning`, so we cap manually
-pytorch_lightning>=2.0,<2.2
+pytorch_lightning>=2.2.2,<2.4
 scipy~=1.10; python_version > "3.7.0"
 scipy~=1.7.3; python_version <= "3.7.0"