From 986f94ee30fe923bbb6f74b562d2581015d3ba9e Mon Sep 17 00:00:00 2001
From: Oleksandr Shchur <shchuro@amazon.com>
Date: Fri, 13 Oct 2023 14:23:02 +0000
Subject: [PATCH] Fix requirements

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

diff --git a/requirements/requirements-pytorch.txt b/requirements/requirements-pytorch.txt
index 9947b4e9ae..03f4e997ab 100644
--- a/requirements/requirements-pytorch.txt
+++ b/requirements/requirements-pytorch.txt
@@ -1,6 +1,7 @@
 torch>=1.9,<3
 lightning>=1.8,<2.2
-lightning.pytorch>=1.8,<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