[deepsparse-ent] setup extra and version check compatibility (#1069) (#1072)

Author: bfineran

setup.py

@@ -58,6 +58,7 @@
 _deepsparse_deps = [
     f"{'deepsparse' if is_release else 'deepsparse-nightly'}~={version_nm_deps}"
 ]
+_deepsparse_ent_deps = [f"deepsparse-ent~={version_nm_deps}"]
 
 _onnxruntime_deps = ["onnxruntime>=1.0.0"]
 _pytorch_deps = [
@@ -117,6 +118,7 @@ def _setup_extras() -> Dict:
     return {
         "dev": _dev_deps,
         "deepsparse": _deepsparse_deps,
+        "deepsparse-ent": _deepsparse_ent_deps,
         "onnxruntime": _onnxruntime_deps,
         "torch": _pytorch_deps,
         "torchvision": _pytorch_vision_deps,

src/sparseml/deepsparse/base.py

@@ -71,7 +71,7 @@ def check_deepsparse_install(
         min_version,
         max_version,
         raise_on_error,
-        alternate_package_names=["deepsparse-nightly"],
+        alternate_package_names=["deepsparse-nightly", "deepsparse-ent"],
     )