From 05a2ab8cd83cdd608bd6e99f3384406155643f58 Mon Sep 17 00:00:00 2001
From: Manuel Pariente <pariente.mnl@gmail.com>
Date: Mon, 6 Dec 2021 10:00:37 +0100
Subject: [PATCH] [install] Fix PL version <1.5.0

---
 requirements/install.txt | 2 +-
 setup.py                 | 2 +-
 2 files changed, 2 insertions(+), 2 deletions(-)

diff --git a/requirements/install.txt b/requirements/install.txt
index da445a4e7..b97e93e52 100644
--- a/requirements/install.txt
+++ b/requirements/install.txt
@@ -2,7 +2,7 @@
 -r ./torchhub.txt
 PyYAML>=5.0
 pandas>=0.23.4
-pytorch-lightning>=1.0.1
+pytorch-lightning>=1.0.1,<1.5.0
 torchaudio>=0.8.0
 pb_bss_eval>=0.0.2
 torch_stoi>=0.0.1
diff --git a/setup.py b/setup.py
index 44b37b81b..e8a635bba 100644
--- a/setup.py
+++ b/setup.py
@@ -46,7 +46,7 @@ def find_version(*file_paths):
         # From requirements/install.txt
         "PyYAML>=5.0",
         "pandas>=0.23.4",
-        "pytorch-lightning>=1.0.1",
+        "pytorch-lightning>=1.0.1,<1.5.0",
         "torchaudio>=0.5.0",
         "pb_bss_eval>=0.0.2",
         "torch_stoi>=0.1.2",