Skip to content
This repository was archived by the owner on Jun 3, 2025. It is now read-only.

Commit 5a5557f

Browse files
authored
Update setup.py version dependencies to use version_base for NM repos (#115)
* Update setup.py version dependencies to use version_base for NM repos * make style
1 parent 8b8412b commit 5a5557f

File tree

1 file changed

+5
-4
lines changed

1 file changed

+5
-4
lines changed

setup.py

Lines changed: 5 additions & 4 deletions
Original file line numberDiff line numberDiff line change
@@ -25,24 +25,25 @@
2525
# default variables to be overwritten by the version.py file
2626
is_release = None
2727
version = "unknown"
28-
version_major_minor = version
28+
version_major = version
29+
version_minor = version
30+
version_bug = version
2931

3032
# load and overwrite version and release info from deepsparse package
3133
version_path = os.path.join("src", "deepsparse", "generated_version.py")
3234
if not os.path.exists(version_path):
3335
version_path = os.path.join("src", "deepsparse", "version.py")
3436
exec(open(version_path).read())
3537
print(f"loaded version {version} from {version_path}")
38+
version_base = f"{version_major}.{version_minor}.{version_bug}"
3639

3740
_PACKAGE_NAME = "deepsparse" if is_release else "deepsparse-nightly"
3841

3942
# File regexes for binaries to include in package_data
4043
binary_regexes = ["*/*.so", "*/*.so.*", "*.bin", "*/*.bin"]
4144

4245
_deps = ["numpy>=1.16.3", "onnx>=1.5.0,<1.8.0", "requests>=2.0.0", "tqdm>=4.0.0"]
43-
_nm_deps = [
44-
f"{'sparsezoo' if is_release else 'sparsezoo-nightly'}~={version_major_minor}"
45-
]
46+
_nm_deps = [f"{'sparsezoo' if is_release else 'sparsezoo-nightly'}~={version_base}"]
4647
_dev_deps = [
4748
"beautifulsoup4==4.9.3",
4849
"black>=20.8b1",

0 commit comments

Comments
 (0)