Replace pytorch-lightning extras dependency (#455)

openvinotoolkit · Jul 27, 2022 · d3aba6e · d3aba6e
1 parent a946140
commit d3aba6e
Showing 1 changed file with 6 additions and 5 deletions.
diff --git 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