Skip to content

Commit

Permalink
[Cherry pick] release version bugfix (#290) (#291)
Browse files Browse the repository at this point in the history
* [Fix] Append date to major_minor version only when
building a nightly release

* missed onnxruntime-gpu
  • Loading branch information
rahul-tuli committed Nov 30, 2023
1 parent 0975ab9 commit 85d29a6
Showing 1 changed file with 7 additions and 6 deletions.
13 changes: 7 additions & 6 deletions setup.py
Original file line number Diff line number Diff line change
Expand Up @@ -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"

Expand All @@ -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 = [
Expand Down Expand Up @@ -73,7 +74,7 @@ def _setup_package_dir() -> Dict:


def _setup_install_requires() -> List:
return _deps
return _nm_deps + _deps


def _setup_extras() -> Dict:
Expand Down

0 comments on commit 85d29a6

Please sign in to comment.