Skip to content
This repository has been archived by the owner on Oct 9, 2023. It is now read-only.

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

2 changes: 1 addition & 1 deletion flash/image/detection/model.py
Original file line number Diff line number Diff line change
Expand Up @@ -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()

Expand Down