From d3aba6e746303309688079311c47a3b8ef0c578b Mon Sep 17 00:00:00 2001
From: Ashwin Vaidya <ashwin.vaidya@intel.com>
Date: Wed, 27 Jul 2022 08:22:16 +0200
Subject: [PATCH] Replace pytorch-lightning extras dependency (#455)

---
 requirements/base.txt | 11 ++++++-----
 1 file changed, 6 insertions(+), 5 deletions(-)

diff --git a/requirements/base.txt b/requirements/base.txt
index 192da4730a..877a51f84d 100644
--- a/requirements/base.txt
+++ b/requirements/base.txt
@@ -1,15 +1,16 @@
 albumentations>=1.1.0
 einops>=0.3.2
-kornia>=0.5.6
+gradio>=2.9.4
+imgaug==0.4.0
+jsonargparse[signatures]>=4.3
+kornia>=0.6.6
+matplotlib>=3.4.3
 omegaconf>=2.1.1
 opencv-python>=4.5.3.56
 pandas>=1.1.0
-pytorch-lightning[extra]>=1.6.0
+pytorch-lightning>=1.6.0
 timm==0.5.4
 torchmetrics>=0.9.0,<=0.9.1
 torchvision>=0.9.1,<=0.12.0
 torchtext>=0.9.1,<=0.12.0
 wandb==0.12.17
-matplotlib>=3.4.3
-gradio>=2.9.4
-imgaug==0.4.0