Add support for Lightning 1.8 + Fixes for the CI

flash/core/integrations/icevision/data.py

@@ -20,7 +20,7 @@
 from flash.core.data.utilities.loading import IMG_EXTENSIONS, load_image, NP_EXTENSIONS
 from flash.core.data.utilities.paths import list_valid_files
 from flash.core.integrations.icevision.transforms import from_icevision_record
-from flash.core.utilities.imports import _ICEVISION_AVAILABLE
+from flash.core.utilities.imports import _ICEVISION_AVAILABLE, requires
 
 if _ICEVISION_AVAILABLE:
     from icevision.core.record import BaseRecord
@@ -33,6 +33,7 @@ class IceVisionInput(Input):
     num_classes: int
     labels: list
 
+    @requires("icevision")
     def load_data(
         self,
         root: str,

requirements.txt

@@ -3,7 +3,7 @@ setuptools<=59.5.0  # Prevent install bug with tensorboard
 numpy
 torch>=1.7.1
 torchmetrics>=0.5.0,!=0.5.1
-pytorch-lightning>=1.3.6
+pytorch-lightning>=1.3.6, <1.8.0
 pyDeprecate
 pandas>=1.1.0
 jsonargparse[signatures]>=3.17.0, <=4.9.0