change Module not found error to a more verbose pip install command (#…

…347)

* change Module not found error to a more verbose pip install command

* change ModuleNotFound error message for tabular

* change ModuleNotFound error message for image_style_transfer

* change ModuleNotFound error message for style transfer fix

* change MoudleNotFound error message for segmentation image

* change ModuleNotFound error message for embedding image

* change ModuleNotFound error message for seq2seq text

* change ModuleNotFound error message for classification text

* Remove -e

* Use \'

* Update error msg

Co-authored-by: Akihiro Nitta <nitta@akihironitta.com>

flash/image/detection/model.py

@@ -94,7 +94,7 @@ def __init__(
     ):
 
         if not _IMAGE_AVAILABLE:
-            raise ModuleNotFoundError("Please, pip install . '[image]'")
+            raise ModuleNotFoundError("Please, pip install 'lightning-flash[image]'")
 
         self.save_hyperparameters()
 

flash/image/embedding/model.py

@@ -60,7 +60,7 @@ def __init__(
         pooling_fn: Callable = torch.max
     ):
         if not _IMAGE_AVAILABLE:
-            raise ModuleNotFoundError("Please, pip install -e '.[image]'")
+            raise ModuleNotFoundError("Please, pip install 'lightning-flash[image]'")
 
         super().__init__(
             model=None,

flash/image/segmentation/data.py

@@ -80,7 +80,7 @@ class SemanticSegmentationPathsDataSource(PathsDataSource):
 
     def __init__(self):
         if not _IMAGE_AVAILABLE:
-            raise ModuleNotFoundError("Please, pip install -e '.[image]'")
+            raise ModuleNotFoundError("Please, pip install 'lightning-flash[image]'")
         super().__init__(IMG_EXTENSIONS)
 
     def load_data(self, data: Union[Tuple[str, str], Tuple[List[str], List[str]]],
@@ -172,7 +172,7 @@ def __init__(
             image_size: A tuple with the expected output image size.
         """
         if not _IMAGE_AVAILABLE:
-            raise ModuleNotFoundError("Please, pip install -e '.[image]'")
+            raise ModuleNotFoundError("Please, pip install 'lightning-flash[image]'")
         self.image_size = image_size
         self.num_classes = num_classes
         if num_classes:

flash/image/segmentation/model.py

@@ -86,7 +86,7 @@ def __init__(
     ) -> None:
 
         if isinstance(backbone, str) and (not _TORCHVISION_AVAILABLE or not _TIMM_AVAILABLE):
-            raise ModuleNotFoundError("Please, pip install -e '.[image]'")
+            raise ModuleNotFoundError("Please, pip install 'lightning-flash[image]'")
 
         if metrics is None:
             metrics = IoU(num_classes=num_classes)

flash/image/style_transfer/model.py

@@ -94,7 +94,7 @@ def __init__(
     ):
 
         if not _IMAGE_STLYE_TRANSFER:
-            raise ModuleNotFoundError("Please, pip install -e '.[image_style_transfer]'")
+            raise ModuleNotFoundError("Please, pip install 'lightning-flash[image_style_transfer]'")
 
         self.save_hyperparameters(ignore="style_image")
 

flash/tabular/classification/model.py

@@ -57,7 +57,7 @@ def __init__(
         **tabnet_kwargs,
     ):
         if not _TABULAR_AVAILABLE:
-            raise ModuleNotFoundError("Please, pip install -e '.[tabular]'")
+            raise ModuleNotFoundError("Please, pip install 'lightning-flash[tabular]'")
 
         self.save_hyperparameters()
 

flash/text/classification/data.py

@@ -36,7 +36,7 @@ def __init__(self, backbone: str, max_length: int = 128):
         super().__init__()
 
         if not _TEXT_AVAILABLE:
-            raise ModuleNotFoundError("Please, pip install -e '.[text]'")
+            raise ModuleNotFoundError("Please, pip install 'lightning-flash[text]'")
 
         self.backbone = backbone
         self.tokenizer = AutoTokenizer.from_pretrained(backbone, use_fast=True)
@@ -203,7 +203,7 @@ def __init__(
     ):
 
         if not _TEXT_AVAILABLE:
-            raise ModuleNotFoundError("Please, pip install -e '.[text]'")
+            raise ModuleNotFoundError("Please, pip install 'lightning-flash[text]'")
 
         self.backbone = backbone
         self.max_length = max_length

flash/text/classification/model.py

@@ -50,7 +50,7 @@ def __init__(
         serializer: Optional[Union[Serializer, Mapping[str, Serializer]]] = None,
     ):
         if not _TEXT_AVAILABLE:
-            raise ModuleNotFoundError("Please, pip install -e '.[text]'")
+            raise ModuleNotFoundError("Please, pip install 'lightning-flash[text]'")
 
         self.save_hyperparameters()
 

flash/text/seq2seq/core/data.py

@@ -43,7 +43,7 @@ def __init__(
         super().__init__()
 
         if not _TEXT_AVAILABLE:
-            raise ModuleNotFoundError("Please, pip install -e '.[text]'")
+            raise ModuleNotFoundError("Please, pip install 'lightning-flash[text]'")
 
         self.backbone = backbone
         self.tokenizer = AutoTokenizer.from_pretrained(self.backbone, use_fast=True)
@@ -236,7 +236,7 @@ def __init__(
         self.padding = padding
 
         if not _TEXT_AVAILABLE:
-            raise ModuleNotFoundError("Please, pip install -e '.[text]'")
+            raise ModuleNotFoundError("Please, pip install 'lightning-flash[text]'")
 
         super().__init__(
             train_transform=train_transform,
@@ -292,7 +292,7 @@ def __init__(self):
         super().__init__()
 
         if not _TEXT_AVAILABLE:
-            raise ModuleNotFoundError("Please, pip install -e '.[text]'")
+            raise ModuleNotFoundError("Please, pip install 'lightning-flash[text]'")
 
         self._backbone = None
         self._tokenizer = None

flash/text/seq2seq/core/model.py

@@ -66,7 +66,7 @@ def __init__(
         num_beams: Optional[int] = None,
     ):
         if not _TEXT_AVAILABLE:
-            raise ModuleNotFoundError("Please, pip install -e '.[text]'")
+            raise ModuleNotFoundError("Please, pip install 'lightning-flash[text]'")
 
         os.environ["TOKENIZERS_PARALLELISM"] = "TRUE"
         # disable HF thousand warnings

flash/text/seq2seq/summarization/metric.py

@@ -60,7 +60,7 @@ def __init__(
     ):
         super().__init__()
         if not _TEXT_AVAILABLE:
-            raise ModuleNotFoundError("Please, pip install -e '.[text]'")
+            raise ModuleNotFoundError("Please, pip install 'lightning-flash[text]'")
 
         self.rouge_newline_sep = rouge_newline_sep
         self.rouge_keys = rouge_keys