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

…been removed (#1564)

* Add import error messages for integrations now that auto install has been removed

* Quality Fixes add # noqa: F401

---------

Co-authored-by: rahul-tuli <rahul@neuralmagic.com>

setup.py

@@ -137,6 +137,7 @@ def _setup_extras() -> Dict:
         "dev": _dev_deps,
         "deepsparse": _deepsparse_deps,
         "deepsparse-ent": _deepsparse_ent_deps,
+        "openpifpaf": _open_pif_paf_deps,
         "onnxruntime": _onnxruntime_deps,
         "torch": _pytorch_deps,
         "torch_all": _pytorch_all_deps,

src/sparseml/openpifpaf/__init__.py

@@ -15,4 +15,12 @@
 from sparseml.analytics import sparseml_analytics as _analytics
 
 
+try:
+    import cv2 as _cv2  # noqa: F401
+
+    import openpifpaf as _openpifpaf  # noqa: F401
+except ImportError:
+    raise ImportError("Please install sparseml[openpifpaf] to use this pathway")
+
+
 _analytics.send_event("python__openpifpaf__init")

src/sparseml/pytorch/torchvision/__init__.py

@@ -15,4 +15,11 @@
 from sparseml.analytics import sparseml_analytics as _analytics
 
 
+try:
+    import cv2 as _cv2  # noqa: F401
+    import torchvision as _torchvision  # noqa: F401
+except ImportError:
+    raise ImportError("Please install sparseml[torchvision] to use this pathway")
+
+
 _analytics.send_event("python__pytorch__torchvision__init")

src/sparseml/transformers/__init__.py

@@ -23,6 +23,13 @@
 from sparseml.analytics import sparseml_analytics as _analytics
 
 
+try:
+    import datasets as _datasets
+    import transformers as _transformers
+except ImportError:
+    raise ImportError("Please install sparseml[transformers] to use this pathway")
+
+
 _analytics.send_event("python__transformers__init")
 
 

src/sparseml/yolov5/__init__.py

@@ -22,6 +22,14 @@
 from sparseml.analytics import sparseml_analytics as _analytics
 
 
+try:
+    import cv2 as _cv2
+    import torchvision as _torchvision
+
+    import yolov5 as _yolov5
+except ImportError:
+    raise ImportError("Please install sparseml[yolov5] to use this pathway")
+
 _analytics.send_event("python__yolov5__init")
 
 _LOGGER = _logging.getLogger(__name__)

src/sparseml/yolov8/__init__.py

@@ -16,6 +16,12 @@
 from sparseml.analytics import sparseml_analytics as _analytics
 
 
+try:
+    import ultralytics as _ultralytics  # noqa: F401
+except ImportError:
+    raise ImportError("Please install sparseml[yolov8] to use this pathway")
+
+
 _analytics.send_event("python__yolov8__init")
 
 if "8.0.30" not in ultralytics.__version__: