diff --git a/setup.py b/setup.py index 52b381dc9a1..9c048140517 100644 --- a/setup.py +++ b/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, diff --git a/src/sparseml/openpifpaf/__init__.py b/src/sparseml/openpifpaf/__init__.py index 1c2637d6480..e038516ad97 100644 --- a/src/sparseml/openpifpaf/__init__.py +++ b/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") diff --git a/src/sparseml/pytorch/torchvision/__init__.py b/src/sparseml/pytorch/torchvision/__init__.py index b127d4c254a..93d36afe43a 100644 --- a/src/sparseml/pytorch/torchvision/__init__.py +++ b/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") diff --git a/src/sparseml/transformers/__init__.py b/src/sparseml/transformers/__init__.py index 2791a8c541b..83cb82e9245 100644 --- a/src/sparseml/transformers/__init__.py +++ b/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") diff --git a/src/sparseml/yolov5/__init__.py b/src/sparseml/yolov5/__init__.py index b3f29dd3247..139735e1c9f 100644 --- a/src/sparseml/yolov5/__init__.py +++ b/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__) diff --git a/src/sparseml/yolov8/__init__.py b/src/sparseml/yolov8/__init__.py index c116eb9ac1a..ae6b61feb6f 100644 --- a/src/sparseml/yolov8/__init__.py +++ b/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__: