Fix version range for lightning (#3023)

awslabs · Oct 20, 2023 · c634ccb · c634ccb
1 parent ec2beb9
commit c634ccb
Showing 1 changed file with 3 additions and 6 deletions.
diff --git a/requirements/requirements-pytorch.txt b/requirements/requirements-pytorch.txt
@@ -1,9 +1,6 @@
 torch>=1.9,<3
-lightning>=1.8,<2.2
+lightning>=2.0,<2.2
 # Capping `lightning` does not cap `pytorch_lightning`, so we cap manually
-pytorch_lightning>=1.8,<2.2
-# Need to pin protobuf (for now)
-# See: https://github.com/PyTorchLightning/pytorch-lightning/issues/13159
-protobuf~=3.19.0
+pytorch_lightning>=2.0,<2.2
 scipy~=1.10; python_version > "3.7.0"
-scipy~=1.7.3; python_version <= "3.7.0"
+scipy~=1.7.3; python_version <= "3.7.0"