From 85d29a69218bc862b77afd1592e04ecd8a32fb0d Mon Sep 17 00:00:00 2001 From: Rahul Tuli Date: Thu, 30 Nov 2023 13:34:13 -0500 Subject: [PATCH] [Cherry pick] release version bugfix (#290) (#291) * [Fix] Append date to major_minor version only when building a nightly release * missed onnxruntime-gpu --- setup.py | 13 +++++++------ 1 file changed, 7 insertions(+), 6 deletions(-) diff --git a/setup.py b/setup.py index 9409b07f..a57ca380 100644 --- a/setup.py +++ b/setup.py @@ -26,7 +26,9 @@ # load and overwrite version and release info from sparseml package exec(open(os.path.join("src", "sparsify", "version.py")).read()) print(f"loaded version {version} from src/sparsify/version.py") -version_nm_deps = f"{version_major_minor}.0.202308" +version_nm_deps = ( + version_major_minor if is_release else f"{version_major_minor}.0.202308" +) _PACKAGE_NAME = "sparsify" if is_release else "sparsify-nightly" @@ -38,13 +40,12 @@ "setuptools>=56.0.0", "optuna>=3.0.2", "onnxruntime-gpu", - f"{'sparsezoo' if is_release else 'sparsezoo-nightly'}~={version_nm_deps}", - f"{'deepsparse' if is_release else 'deepsparse-nightly'}~={version_nm_deps}", - f"{'sparseml' if is_release else 'sparseml-nightly'}[torchvision,yolov5]~={version_nm_deps}", # noqa E501 ] _nm_deps = [ - f"{'sparseml' if is_release else 'sparseml-nightly'}[transformers]~={version_nm_deps}", # noqa E501 + f"{'sparsezoo' if is_release else 'sparsezoo-nightly'}~={version_nm_deps}", + f"{'deepsparse' if is_release else 'deepsparse-nightly'}~={version_nm_deps}", + f"{'sparseml' if is_release else 'sparseml-nightly'}[torchvision,yolov5]~={version_nm_deps}", # noqa E501 ] _dev_deps = [ @@ -73,7 +74,7 @@ def _setup_package_dir() -> Dict: def _setup_install_requires() -> List: - return _deps + return _nm_deps + _deps def _setup_extras() -> Dict: