bump pytorch lightning compat

awslabs · Jun 14, 2024 · c136c8b · c136c8b
1 parent f0f2266
commit c136c8b
Showing 1 changed file with 2 additions and 2 deletions.
diff --git 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"