Add import error messages for integrations now that auto install has …

…been removed (#1030)

(cherry picked from commit 4e03e41)

src/deepsparse/image_classification/__init__.py

@@ -16,9 +16,19 @@
 from deepsparse.analytics import deepsparse_analytics as _analytics
 
 
-_analytics.send_event("python__image_classification__init")
+try:
+    import torchvision as _torchvision
+
+    import cv2 as _cv2
+except ImportError:
+    raise ImportError(
+        "Please install deepsparse[image_classification] to use this pathway"
+    )
 
 
 from .constants import *
 from .pipelines import *
 from .schemas import *
+
+
+_analytics.send_event("python__image_classification__init")

src/deepsparse/open_pif_paf/__init__.py

@@ -14,6 +14,13 @@
 # flake8: noqa
 from deepsparse.analytics import deepsparse_analytics as _analytics
 
+
+try:
+    import cv2 as _cv2
+    import openpifpaf as _openpifpaf
+except ImportError:
+    raise ImportError("Please install deepsparse[openpifpaf] to use this pathway")
+
 from .utils import *
 
 

src/deepsparse/server/__init__.py

@@ -21,6 +21,14 @@
 
 from deepsparse.analytics import deepsparse_analytics as _analytics
 
+
+try:
+    import uvicorn as _uvicorn
+    from fastapi import FastAPI as _FastAPI
+    from starlette.responses import RedirectResponse as _RedirectResponse
+except ImportError:
+    raise ImportError("Please install deepsparse[server] to use this pathway")
+
 from .cli import main
 
 

src/deepsparse/transformers/__init__.py

@@ -27,6 +27,14 @@
 _analytics.send_event("python__transformers__init")
 
 
+try:
+    import transformers as _transformers
+
+    import datasets as _datasets
+except ImportError:
+    raise ImportError("Please install deepsparse[transformers] to use this pathway")
+
+
 _LOGGER = _logging.getLogger(__name__)
 
 

src/deepsparse/yolo/__init__.py

@@ -16,6 +16,14 @@
 
 from deepsparse.analytics import deepsparse_analytics as _analytics
 
+
+try:
+    import torchvision as _torchvision
+
+    import cv2 as _cv2
+except ImportError:
+    raise ImportError("Please install deepsparse[yolo] to use this pathway")
+
 from .annotate import *
 from .pipelines import *
 from .schemas import *

src/deepsparse/yolov8/__init__.py

@@ -16,6 +16,13 @@
 
 from deepsparse.analytics import deepsparse_analytics as _analytics
 
+
+try:
+    import ultralytics as _ultralytics
+except ImportError:
+    raise ImportError("Please install deepsparse[yolov8] to use this pathway")
+
+
 from .annotate import *
 from .pipelines import *
 from .schemas import *